문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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] |
입출력 예 설명
입출력 예 #1
- 수포자 1은 모든 문제를 맞혔습니다.
- 수포자 2는 모든 문제를 틀렸습니다.
- 수포자 3은 모든 문제를 틀렸습니다.
따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.
입출력 예 #2
모든 사람이 2문제씩을 맞췄습니다.
풀이 방식
이 문제는 답이 주어졌을 때 최다 문제를 맞춘 학생을 구하는 문제입니다.
저는 아래와 같은 알고리즘으로 이 문제를 풀었습니다.
- 각 학생별 맞춘 문제 개수를 구합니다.
- 가장 많이 맞춘 문제 개수를 구합니다.
- 문제를 가장 많이 맞춘 학생의 수를 구한 뒤, 답을 넣을 배열을 해당 수만큼 잡습니다.
- 만약 어떤 학생의 가장 많이 맞춘 개수가 최댓값과 같다면 그 학생을 정답 배열에 넣은 뒤 정답 배열의 인덱스 값을 증가시킵니다.
그럼 더 자세한 풀이를 볼까요?
public int[] solution(int[] answers) {
int[] score = new int[3];
int[] a = {1,2,3,4,5};
int[] b = {2,1,2,3,2,4,2,5};
int[] c = {3,3,1,1,2,2,4,4,5,5};각 학생별로 맞춘 문제의 개수를 저장할 score배열을 학생 수만큼 잡아줍니다.
학생 a b c별로 반복되는 답이 있으므로 반복되는 부분만을 제외한 답을 각각의 배열에 담아줍니다.
for ( int i = 0; i < answers.length; i++) {
if ( a[i%a.length] == answers[i])
score[0]++;
if ( b[i%b.length] == answers[i])
score[1]++;
if ( c[i%c.length] == answers[i])
score[2]++;
}각 학생들이 답을 맞췄는지를 확인합니다.
%를 사용하는 이유는 학생들이 적은 답이 특정 구간마다 반복이 되기 때문에 사용하였습니다.
또한 score배열에서는 순서대로 1 2 3학생이므로 인덱스는 정해줍니다.
int max = Math.max(score[0], Math.max(score[1], score[2]));학생별 맞춘 답 개수의 최댓값을 구합니다.
메서드로는 Math클래스의 max()를 사용하였으며,
최다 답을 맞춘 사람을 판별하기 위한 절차입니다.
int cnt = 0;
for ( int i = 0; i < score.length; i++) {
if ( score[i] == max)
cnt++;
}가장 많은 문제를 맞춘 사람의 명수를 구합니다.
앞서 구했던 학생별 맞춘 답 개수가 최댓값과 같다면 가장 많이 문제를 맞춘 사람의 명수를 1 증가시킵니다.
int[] answer = new int[cnt];
for ( int i = 0, j = 0; i < score.length; i++) {
if ( score[i] == max ) {
answer[j++] = i+1;
}
}
return answer;answer배열에 가장 많은 문제를 맞춘 학생을 저장합니다.
앞서 구했던 cnt를 길이로 가지는 answer배열을 만들어줍니다.
이 answer배열에는 문제를 가장 많이 맞춘 사람을 넣을 것입니다.
만약 학생의 맞춘 답 개수가 최댓값과 같다면 그 학생이 바로 가장 많은 문제를 맞춘 학생이므로 answer배열에 i+1를 저장해줍니다.
이는 i는 0부터 시작하기 때문에 1부터 시작하는 학생을 맞추기 위함입니다.
저장한 뒤에 인덱스값을 증가시켜 다음 학생이 있을 경우 다음 인덱스칸에 저장될 수 있도록 조치합니다.
모든 연산이 끝난 뒤 answer을 리턴해주어 문제를 해결합니다.
소스 코드
import java.util.Arrays;
class Solution {
public int[] solution(int[] answers) {
int[] score = new int[3];
int[] a = {1,2,3,4,5};
int[] b = {2,1,2,3,2,4,2,5};
int[] c = {3,3,1,1,2,2,4,4,5,5};
for ( int i = 0; i < answers.length; i++) {
if ( a[i%a.length] == answers[i])
score[0]++;
if ( b[i%b.length] == answers[i])
score[1]++;
if ( c[i%c.length] == answers[i])
score[2]++;
}
int max = Math.max(score[0], Math.max(score[1], score[2]));
int cnt = 0;
for ( int i = 0; i < score.length; i++) {
if ( score[i] == max)
cnt++;
}
int[] answer = new int[cnt];
for ( int i = 0, j = 0; i < score.length; i++) {
if ( score[i] == max ) {
answer[j++] = i+1;
}
}
return answer;
}
}실행 결과
