내용 복습/알고리즘

프로그래머스 lv.1 모의고사

jskim4695 2024. 9. 27. 17:37

이 문제는 완전탐색 문제이다.

 

문제 설명

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

 

function solution(answers) {
    var answer = [];
    const score = []
    var giveup1 = [1,2,3,4,5]
    var giveup2 = [2,1,2,3,2,4,2,5]
    var giveup3 = [3,3,1,1,2,2,4,4,5,5]
    
    const countAnswer = (studentAnswer) => {
        let checkAnswer = [...answers]
        let count = 0
        
        while (checkAnswer.length !== 0) {
            const checkRange = checkAnswer.slice(0, studentAnswer.length)
            
            const isSameValue = (checkRange, studentAnswer) => checkRange.filter((v,i) => v===studentAnswer[i])
            
            const countArray = isSameValue(checkRange, studentAnswer)
            count += countArray.length
            checkAnswer.splice(0, studentAnswer.length)
        }
        return count
    }
    
    score.push(countAnswer(giveup1))
    score.push(countAnswer(giveup2))
    score.push(countAnswer(giveup3))
    
    score.forEach((v,i) => {
        let maxValue = Math.max(...score)
        if (v === maxValue) {
            answer.push(i + 1)
        }
    })
    
    return answer;
}

 

이전에 내가 풀었던 답안이다. 보다시피 너무 풀어 적어서 엄청 길고 비효율적인 코드가 많다. 이번에 코딩테스트 합격자되기 서적을 구매하면서 배운 방법으로 풀이하려고 한다.

 

function solution(answers) {
    const patterns = [
        [1, 2, 3, 4, 5], // 1번 수포자 패턴
        [2, 1, 2, 3, 2, 4, 2, 5], // 2번 수포자 패턴
        [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] // 3번 수포자 패턴
    ]
    
    const scores = [0, 0, 0];
    
    for (const [i, answer] of answers.entries()) {
        for (const [j, pattern] of patterns.entries()) {
            if (answer === pattern[i % pattern.length]) {
                scores[j] += 1;
            }
        }
    }
   
    // 가장 높은 점수 저장
    const maxScore = Math.max(...scores);
    
    // 가장 높은 점수를 받은 수포자들의 번호를 찾아서 배열에 담음
    const highestScores = [];
    for (let i = 0; i<scores.length; i++) {
        if (scores[i] === maxScore) {
            highestScores.push(i + 1);
        }
    }
    return highestScores;
}

 

배열의 entries() 메서드는 배열 내 데이터를 인덱스 값으로 묶어서 순회할 수 있는 Iterator 객체를 반환한다. Iterator는 next() 메서드를 이용하여 연속된 데이터를 순회할 수 있게 만드는 객체다.

of 연산자를 이용하면 처음부터 끝까지 순회하는 것이 가능하며 i는 answer의 인덱스이고 answer은 해당 인덱스의 실제 값이다.

이렇게 배열 전체를 반복해야 하는 경우 of 연산자를 활용하면 편리하게 반복문을 작성할 수 있다.

 

바깥 반복문은 answers, 안쪽 반복문은 patterns의 데이터를 하나씩 가리키고 각 답안과 수포자의 값을 하나씩 비교하면서 정답이 일치하는 수포자의 경우 scores를 1만큼 더한다. 정답 패턴의 길이가 수포자의 답안 길이보다 긴 경우 정답 패턴의 처음 데이터와 다시 비교할 수 있도록 나머지 연산자를 사용했다.

무작정 문제를 해석해 풀기보단 이렇식으로 효율적인 코드가 좋은 것 같아 정리해보았다.