Cos Pro 1급 - 샘플 문제 풀이 1차 5번 (배열의 원소의 갯수 세기)

문제 지문 #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 값 설명

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

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

---

예시

N	votes	return
5	[1,5,4,3,2,5,2,5,5,4]	[5]
4	[1,3,2,3,2]	[2,3]

예시 설명 #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 하면 됩니다.

---

혼자 풀이

이번 문제 유형은 메서드 전체 작성, 빈칸 채우기에 이어 한 줄 고치기입니다. 
즉, 디버깅을 하는 문제인데요. 저는 처음에 눈으로 로직 따라가며 살펴보았는데, 한 번에 틀린 곳을 찾지는 못했습니다.
디버그 모드에서 변수값 따라가면서 문제를 확인하고 쉽게 풀었습니다. 실제 시험에서도 그렇게 해야 될 것 같네요.
(내용 정정합니다. 실제 시험에서는 intellij나 eclipse를 사용하지 못하며, 디버깅 모드로 실행이 안되니, 머릿속으로 로직을 따라가는 훈련을 해야 합니다.)

알고리즘은 간단합니다. 투표 건을 순차적으로 for문 수행하면서 voteCounter 배열에 후보자 번호를 인덱스로 입력해줍니다. 그리고 최다 득표자가 2명 이상이면, cnt를 증가시킵니다. 최종적으로 최다 득표 후보자의 번호를 배열에 담아 리턴하면 됩니다. voteCounter의 인덱스가 후보자 번호이므로 오름차순 정렬하는 부분은 따로 신경 쓰지 않아도 됩니다.

틀린 부분은 후보자의 번호를 배열에 담아 리턴해야 되는데, 득표수를 배열에 담아 리턴하는 부분이었습니다.

import java.util.*;

class MySolution {
    public int[] solution(int N, int[] votes) {
        int voteCounter[] = new int[11];
        for (int i = 0; i < votes.length; i++) {
            voteCounter[votes[i]] += 1;
        }
        int maxVal = 0;
        int cnt = 0;
        for (int i = 1; i <= N; i++) {
            if (maxVal < voteCounter[i]) {
                maxVal = voteCounter[i];
                cnt = 1;
            }
            else if(maxVal == voteCounter[i]){
                cnt += 1;
            }
        }
        int answer[] = new int[cnt];
        for (int i = 1, idx = 0; i <= N; i++){
            if (voteCounter[i] == maxVal) {
//            	answer[idx] = voteCounter[i];
                answer[idx]=i;   //수정한 코드
                idx += 1;
            }
        }
        return answer;
    }

    // The following is main method to output testcase. The main method is correct and you shall correct solution method.
    public static void main(String[] args) {
        MySolution sol = new MySolution();
        int N1 = 5;
        int[] votes1 = {1,5,4,3,2,5,2,5,5,4};
        int[] ret1 = sol.solution(N1, votes1);

        // Press Run button to receive output.
        System.out.println("Solution: return value of the method is " + Arrays.toString(ret1) + " .");


        int N2 = 4;
        int[] votes2 = {1, 3, 2, 3, 2};
        int[] ret2 = sol.solution(N2, votes2);

        // Press Run button to receive output.
        System.out.println("Solution: return value of the method is " + Arrays.toString(ret2) + " .");
    }
}

 

첨부 파일

프로젝트 파일 전체를 첨부합니다.
Solution 이 문제, CorrectSolution 이 정답, MySolution이 제가 푼 것입니다.
"Project from existing sources..." 메뉴에서 불러다 쓰시면 됩니다.

FirstQuestion5.zip

0.01MB

출처: https://www.ybmit.com/cos_pro/cos_pro_r_test.jsp