[COS Pro 2급, Python] 2차 9번 : 투표에 대한 후보 찾기

N명의 후보에 대해 투표한 결과가 들어있는 배열이 있습니다.

예를 들어 5명의 후보에 대해 투표를 진행한 결과가 [2, 5, 3, 4, 1, 5, 1, 5, 5, 3]이라면 순서대로 [2번, 5번, 3번, 4번, 1번, 5번, 1번, 5번, 5번, 3번] 후보에 투표했음을 나타냅니다.

이때, 정확히 K 표를 받은 후보는 총 몇 명인지 구하려 합니다. 

 

예를 들어 K = 2일 때, 위 투표 결과에서 정확히 2표를 받은 후보는 1번, 3번 후보로, 총 2명입니다. 

 

투표 결과가 들어있는 배열 votes와 votes의 길이 votes_len, 후보의 수 N, 표의 개수 K가 매개변수로 주어질 때, K 표를 받은 후보는 몇 명인지 return 하도록 solution 함수를 작성했습니다.

그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다.

주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

 

 

code

def solution(votes, N, K):
	counter = [0 for _ in range(N + 1)]
	for x in votes:
		counter[x] += 1
	answer = 0
	for c in counter:
		if c == K:
			answer += 1
	return answer
    
    
votes = [2, 5, 3, 4, 1, 5, 1, 5, 5, 3]
N = 5
K = 2
ret = solution(votes, N, K)

print("solution 함수의 반환 값은", ret, "입니다.")