위 문제는 YBM COS-Pro의 1차 1급 5번 문제에 관한 설명입니다.

문제

1번부터 N번까지 후보에 대해서 투표를 진행했습니다.

예를 들어 투표 결과가 [1, 5, 4, 3, 2, 5, 2, 5, 5, 4]라면 순서대로 [1번, 5번, 4번, 3번, 2번, 5번, 2번, 5번, 5번, 4번] 후보에

투표했음을 나타냅니다. 이때, 가장 많은 표를 받은 후보의 번호를 구하려고 합니다.

주어진 solution 메소드는 후보의 수 N과 투표를 진행한 결과가 담긴 배열 votes가 매개변수로 주어졌을 때,

가장 많은 표를 받은 후보의 번호를 return 하는 메소드입니다. 그러나, 코드 일부분이 잘못되어있기 때문에,

몇몇 입력에 대해서는 올바르게 동작하지 않습니다.

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

입력

후보의 수 N과 투표 결과가 들어있는 배열 votes가 solution 메소드의 매개변수로 주어집니다.

• N은 1 이상 10 이하의 자연수입니다.
• votes의 길이는 1 이상 100 이하입니다.
• votes의 원소는 1 이상 N이하의 자연수입니다.

출력

가장 많은 표를 받은 후보의 번호를 배열에 담아 return 해주세요

• 만약 가장 많은 표를 받은 후보가 2개 이상이라면, 그 후보들의 번호를 모두 배열에 담아 오름차순 정렬하여 return 해주세요.

예시

예시 설명

예시 #1

1번부터 5번까지 5개의 후보가 있으며, 투표 결과는 [1, 5, 4, 3, 2, 5, 2, 5, 5, 4]입니다. 각 후보의 득표수는 다음과 같습니다.

• 1번 후보가 1표
• 2번 후보가 2표
• 3번 후보가 1표
• 4번 후보가 2표
• 5번 후보가 4표

이 경우 5번 후보가 4표로 가장 많은 표를 얻었습니다.

예시 #2

1번 후보가 1표, 2번 후보가 2표, 3번 후보가 2표씩 받았습니다. 2번과 3번 후보가 공동으로 가장 많은 표를 받았기 때문에 [2, 3]을 오름차순 정렬하여 return 하면 됩니다.

class Main{
    public int[] solution(int N, int[] votes) {
        // N = 후보자수 , votes[] = 투표를 진행한 결과
        int vote_counter[] = new int[11];
        for(int i =0 ; i < votes.length; i++)
        {
            vote_counter[votes[i]]+=1; // votes에 있는 투표결과를 count
        }
        int max_val = 0;
        int cnt = 0;
        for(int i =1; i<= N;i++)
        {
            if(max_val < vote_counter[i]){
                max_val = vote_counter[i];
                cnt = 1; // 최대값 후보 갱신
            }
            else if(max_val == vote_counter[i])
            {
                cnt += 1; // 후보군 수 count
            }
        }
        int answer[] = new int [cnt];
        for(int i =1, index = 0; i <= N; i++ ) //앞 번호 순서대로 정렬, 배열에 옮기기
            if(vote_counter[i]== max_val)
            {
                answer[index]= i;
                index += 1;
            }
        return answer;
    }
}

코드 설명

1. votes[]에 들어있는 투표 결과를 vote_counter를 통해 각 번호마다의 득표 수를 count합니다.

2. 최대값을 구할 때 현재가지고 있는 max_val보다 큰 값이 있다면 그값을 max_val로 넣고 cnt = 1로 “최대값 후보는 1명이 있다.”를 표현합니다.

3. 만약 max_val와 같은 득표 수를 가진 사람이 더 있다면 cnt를 늘려 “후보값의 수가 다수” 임을 표현합니다.

4. answer배열에 앞 번호순서대로 최대 득표받은 번호군을 옮겨줍니다.