[프로그래머스] 모의고사

🗒️ 문제

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.

📝 문제 해설

정말 쉬운 문제라고 생각했다. 먼저 수포자 인원이 3명밖에 없기 때문에 3명의 패턴을 담은 배열을 각각 만들어주었고, 정답과 비교하기 위해서 카운트를 담는 변수도 각각 만들어주었다. 정답을 담은 배열의 길이는 10,000개까지 가능하지만 수포자의 배열의 길이는 정해져 있기 때문에 이 둘의 배열을 비교하기 위해서 나머지 연산을 사용하였다.

cnt1, cnt2, cnt3 각각 카운트를 담은 배열을 만들어주고 Stack을 사용하여 cnt 값보다 크면 pop과 push를 하고 작으면 continue하고, 같으면 이어서 push만 하는 연산을 진행하였다.

하지만 생각보다 오류가 많이 발생하였다. 내가 작성했던 코드에서 문제점은 예를 들어 cnt로 들어온 값이 [2, 2, 5]라면 result라는 배열에 [2,5]가 저장이 되어 버린다. pop()연산을 한번만 하기 때문에 이런 예외가 발생한다. 이 문제를 해결하기 위해서는 배열을 빈배열로 만들어주고 큰 값을 push해주면 해결이 된다.

💡 필요 문법

push()

배열 가장 끝에 값을 추가한다.

pop()

배열 가장 끝 값을 꺼낸다.

💻 코드

function solution(answers) {
    // 수포자 1, 2, 3의 패턴이 담긴 배열 생성
    let N1 = [1,2,3,4,5];
    let N2 = [2,1,2,3,2,4,2,5];
    let N3 = [3,3,1,1,2,2,4,4,5,5];
    
    // 수포자 1, 2, 3의 정답 카운트를 담을 변수 생성
    let cnt1 = 0;
    let cnt2 = 0;
    let cnt3 = 0;
    
    // 배열의 원소를 하나하나 비교해서 정답 카운트를 Update
    for(let i=0; i<answers.length; i++){
        if(answers[i] === N1[i%5]) cnt1++;
        if(answers[i] === N2[i%8]) cnt2++;
        if(answers[i] === N3[i%10]) cnt3++;
    }
    
    // 카운트가 가장 많은 수포자를 배열에 넣어서 리턴 
    let arr = [cnt1, cnt2, cnt3];
    let result = [];
    let k = -1;
    
    for(let i=0; i<arr.length; i++){
        if(arr[i] > k){
            result = [];
            result.push(i+1);
            k = arr[i];
        }
        else if(arr[i] === k){
            result.push(i+1);
        }
        continue;
    }
    return result;
}