[COS PRO 2급-Python] 투표에 대한 후보 찾기
문제 설명
N명의 후보에 대해 투표한 결과가 들어있는 배열이 있습니다.
예를 들어 5명의 후보에 대해 투표를 진행한 결과가 [2, 5, 3, 4, 1, 5, 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 함수를 작성했습니다.
그러나, 코드 일부분이 잘못되어있기 때무에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다.
주어진 코드를 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.
매개변수 설명
투표 겨로가가 들어있는 배열 votes와 votes의 길이 votes_len, 후보의 수 N, 표의 개수 K가 solution 함수의 매개변수로 주어집니다.
- votes_len은 10 이상, 100 이하의 자연수입니다.
- votes의 원소느 1 이상, 전체 후보의 수 N 이하의 자연수입니다.
- N은 3 이상 10 이하의 자연수입니다.
- k는 0 이상 100 이하의 정수입니다.
return 값 설명
K 표를 받은 후보는 몇 명인지 return 해주세요.
예시
알고리즘 구상
▶ votes의 있는 값을 하나씩 꺼내와서 값은 값이 나올 때 마다 counter의 x번째 배열에 +1을 한다!
내가 쓴 코드!
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느낀점
파이썬의 배열도 다양한 것 같다! 배열과 리스트에 대해 더 공부를 해보자!
