(Python, Java) 프로그래머스 - 모의고사

31 Dec 2021 in Algorithm on Programmers, Level1

[문제 링크]

Python 풀이

def solution(answers):
    answer = []
    length = len(answers)
    students = [[1, 2, 3, 4, 5], [2, 1, 2, 3, 2, 4, 2, 5], [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
    correct = [0, 0, 0]
    length1 = len(students[0])
    length2 = len(students[1])
    length3 = len(students[2])
    for idx in range(length):
        if answers[idx] == students[0][idx % length1]:
            correct[0] += 1

        if answers[idx] == students[1][idx % length2]:
            correct[1] += 1

        if answers[idx] == students[2][idx % length3]:
            correct[2] += 1

    top = max(correct)
    for idx in range(3):
        if top == correct[idx]:
            answer.append(idx + 1)

    return answer

Java 풀이

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] answers) {
        ArrayList<Integer> answer = new ArrayList<>();

        int length = answers.length;
        int[][] value =  {{1, 2, 3, 4, 5}, {2, 1, 2, 3, 2, 4, 2, 5}, {3, 3, 1, 1, 2, 2, 4, 4, 5, 5}} ;

        int[] count = new int[3];

        for (int i = 0; i < length; i++) {
            for (int j = 0; j < 3; j++) {
                int idx = i % value[j].length;
                if (answers[i] == value[j][idx]) {
                    count[j] += 1;
                }
            }
        }
        int maxCount = Arrays.stream(count).max().getAsInt();
        for (int i = 0; i < 3; i++) {
            if (count[i] == maxCount)
                answer.add(i + 1);
        }

        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}