[코딩테스트] 주사위 게임 문제 풀이 (빈도 계산)

주사위 게임 문제

*프로그래머스 발췌*

1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다.
네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.
• 네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
• 세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)2 점을 얻습니다.
• 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
• 어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
• 네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.
네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

문제 풀이1 - Map

function solution(a, b, c, d) {
    let map = new Map();

    // 주사위 결과를 Map에 저장 (숫자: 등장 횟수)
    [a, b, c, d].forEach((el) => {
        map.set(el, map.get(el) ? map.get(el) + 1 : 1)
    });

    // 모든 숫자가 같은 경우
    if (map.size === 1) {
        const [p] = map.keys();
        return 1111 * p;
    }

    // 세 숫자가 같고, 나머지 하나가 다른 경우
    if (map.size === 2) {
        const [p, q] = [...map.keys()];
        const pCount = map.get(p);
        const qCount = map.get(q);

        if (pCount === 3) return (10 * p + q) ** 2;
        if (qCount === 3) return (10 * q + p) ** 2;
        
        // 두 숫자가 각각 두 번 나온 경우
        return (p + q) * Math.abs(p - q);
    }

    // 두 숫자가 같고 나머지 두 숫자가 다른 경우
    if (map.size === 3) {
        let singleKeys = [];
        map.forEach((value, key) => {
            if (value === 1) singleKeys.push(key);
        });
        return singleKeys[0] * singleKeys[1];
    }

    // 모든 숫자가 다른 경우
    if (map.size === 4) {
        return Math.min(...map.keys());
    }
}

 

문제 풀이2

function solution(a, b, c, d) {
    const dice = [a, b, c, d];
    
    // 주사위 값의 빈도 계산
    const counts = {};
    dice.forEach(num => {
        counts[num] = (counts[num] || 0) + 1;
    });

    // 고유한 숫자와 그 빈도의 배열
    const uniqueNumbers = Object.keys(counts).map(Number);
    const frequencies = Object.values(counts);

    if (uniqueNumbers.length === 1) {
        // 모든 숫자가 같음
        return 1111 * uniqueNumbers[0];
    } else if (uniqueNumbers.length === 2) {
        if (frequencies.includes(3)) {
            // 3개의 숫자가 같음
            const p = uniqueNumbers[frequencies.indexOf(3)];
            const q = uniqueNumbers[frequencies.indexOf(1)];
            return (10 * p + q) ** 2;
        } else {
            // 2개씩 같은 숫자가 나옴
            const [p, q] = uniqueNumbers;
            return (p + q) * Math.abs(p - q);
        }
    } else if (uniqueNumbers.length === 3) {
        // 2개의 숫자가 같고 나머지 2개가 다름
        const [q, r] = uniqueNumbers.filter(num => counts[num] === 1);
        return q * r;
    } else {
        // 모든 숫자가 다름
        return Math.min(...dice);
    }
}