[완전탐색] 모의고사 (JAVA)

# 문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...

2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...

3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

---

제한 조건

• 시험은 최대 10,000 문제로 구성되어있습니다.
• 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

---

입출력 예

answers return

[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

 

# 정답 코드

public int[] solution(int[] answers) {
        //수포자 3명
        ArrayList<Integer> one = new ArrayList<>(Arrays.asList(1,2,3,4,5));
        ArrayList<Integer> two = new ArrayList<>(Arrays.asList(2,1,2,3,2,4,2,5));
        ArrayList<Integer> three = new ArrayList<>(Arrays.asList(3,3,1,1,2,2,4,4,5,5));
        
        //수포자 각각의 인덱스
        int cnt1 = 0;
        int cnt2 = 0;
        int cnt3 = 0;
        
        int[] score = {0,0,0}; //점수 배열
        
        //점수 계산하기
        for (int i = 0; i < answers.length; i++) {
        	if (answers[i] == one.get(cnt1)) {
        		score[0]++;
        	}
        	if (answers[i] == two.get(cnt2)) {
        		score[1]++;
        	}
        	if (answers[i] == three.get(cnt3)) {
        		score[2]++;
        	}
        	
        	cnt1 = (cnt1 + 1) % one.size();
        	cnt2 = (cnt2 + 1) % two.size();
        	cnt3 = (cnt3 + 1) % three.size();
        }
        
        int max = score[0]; //점수 최대값
        ArrayList<Integer> answerList = new ArrayList<>(Arrays.asList(1)); //리턴값
        
        //가장 높은 점수를 받은 사람 찾기
        for (int i = 1; i < score.length; i++) {
        	if (score[i] > max) {
        		answerList.removeAll(answerList);
        		answerList.add(i+1);
        		max = score[i];
        	}
        	else if (score[i] == max) {
        		answerList.add(i+1);
        	}
        }
        
        //int 배열로 바꿔 리턴
        int[] answer = new int[answerList.size()];
        
        for (int i = 0; i < answerList.size(); i++) {
        	answer[i] = answerList.get(i);
        }
        
        return answer;
    }

 

레벨 1 문제인데 이렇게 푸는게 맞나 싶다... 더 짧은 코드가 있을 것 같다.

 

수포자가 찍는 방식은 정해져 있으므로 ArrayList로 각각 생성한다.

각 ArrayList의 길이가 다르기 때문에 각각 인덱스를 생성한다.

점수를 저장하기 위해 int 배열을 사용한다.

answers를 순회하면서 점수를 계산한다.

이때 cnt1, cnt2, cnt3은 answers만큼 계속 반복되어야 한다.

따라서 각 배열을 원형큐로 사용할 수 있게 갱신해준다.

 

점수 최대값을 저장하기 위해 max를 사용한다.

점수 배열 score를 순회하면서 가장 높은 점수를 받은 사람을 찾는다.

answerList에 첫번째 사람의 점수를 넣어놓고 순회하면서 다른 사람의 점수와 비교한다.

• 다른 사람이 더 높으면, answerList를 비우고 현재 사람을 추가한다.
• 점수가 같으면, 현재 사람을 추가한다.

마지막으로 ArrayList인 answerList를 int 배열로 바꾸고 리턴한다.