Python 짝수 홀수 개수 [코딩테스트 입문 - Day6]

---

문제

---

구현

num_list를 돌며 홀수, 짝수의 개수를 각각 카운트한다.

---

코드

def solution(num_list):
    odd, even = 0, 0
    for n in num_list:
        if n%2==0:
            even+=1
        else:
            odd+=1
    return [even, odd]

 

---

다른 풀이

def solution(num_list):
    answer = [0,0]
    for n in num_list:
        answer[n%2]+=1
    return answer

위의 풀이에서 조금만 더 생각하면 %2 연산을 통해 answer의 홀수, 짝수값에 넣어주도록 할 수 있다.

 

---

또 다른 풀이

def solution(num_list):
    odd = sum(1 for n in num_list if n % 2)
    return [len(num_list) - odd, odd]

list comprehension을 통해 짝수의 개수만 구하고 홀수는 전체 길이에서 빼주면 된다.

하지만 이 방법은 약간의 계산 코스트가 더 들어간다.