문제 이름 모의고사
문제 출처 https://school.programmers.co.kr/learn/courses/30/lessons/42840
문제
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.
내 풀이
import java.util.ArrayList;
import java.util.Arrays;
class Solution {
public int[] solution(int[] answers) {
ArrayList<Integer> result = new ArrayList<>();
int[] first = new int[]{1, 2, 3, 4, 5};
int[] second = new int[]{2, 1, 2, 3, 2, 4, 2, 5};
int[] third = new int[]{3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
int[] scores = new int[]{0, 0, 0};
for (int i = 0; i < answers.length; i++) {
if (first[i % first.length] == answers[i]) {
scores[0] += 1;
}
if (second[i % second.length] == answers[i]) {
scores[1] += 1;
}
if (third[i % third.length] == answers[i]) {
scores[2] += 1;
}
}
int maxScore = Arrays.stream(scores).max().getAsInt();
for (int i = 0; i < 3; i++) {
if (scores[i] == maxScore) {
result.add(i + 1);
}
}
return result.stream().mapToInt(Integer::intValue).toArray();
}
}코드 설명
각 수포자가 정답을 찍는 패턴을 배열로 저장해, 최고 점수를 구한다.
이후 점수 목록을 순회하며 최고 점수와 같다면 ArrayList에 저장한다.
마지막으로 ArrayList를 배열로 바꿔서 반환하였다.
문제에서는 동점자가 있다면 오름차순으로 정렬해 반환하라고 되어있는데, for문에서 1, 2, 3번 수포자를 오름차순으로 순회해 저장하고 있기 때문에 따로 재정렬은 하지 않았다.
MEMO
최고 점수를 맞은 사람의 인덱스를 어떻게 가져와야할지 몰라서 답안을 조금 보고 작성했다.
stream을 이용한 코드가 많아서 최고값의 인덱스도 그렇게 팬시하게 가져올 수 있을 줄 알았는데 그런건 없었다 ㅎㅎ
풀이 코드
import java.util.ArrayList;
import java.util.Arrays;
class Solution{
public int[] solution(int[] answers) {
int[][] pattern = new int[][]{{1, 2, 3, 4, 5},
{2, 1, 2, 3, 2, 4, 2, 5},
{3, 3, 1, 1, 2, 2, 4, 4, 5, 5}};
int[] scores = new int[3];
for (int i = 0; i < answers.length; i++) {
for(int j = 0; j < pattern.length; j++){
if(answers[i] == pattern[j][i % pattern[j].length]){
scores[j]++;
}
}
}
int maxScore = Arrays.stream(scores).max().getAsInt();
ArrayList<Integer> answer = new ArrayList<>();
for (int i = 0; i < 3; i++) {
if (scores[i] == maxScore) {
answer.add(i + 1);
}
}
return answer.stream().mapToInt(Integer::intValue).toArray();
}
}
출처 : 책 '코딩테스트 합격자되기'
내 코드와의 대표적인 차이점은 찍기 패턴을 이중 리스트로 선언하고, for문에서 이를 이용한 것인것 같다.
만약 수포자의 수가 많았다면 나도 비슷한 코드를 썼겠지만, 3명밖에 없기 때문에 직접 써주는 게 가독성과 실수를 줄일 수 있을 것이라 판단했다.
