LeetCode [JS] > 1128. Number of Equivalent Domino Pairs

문제링크

https://leetcode.com/problems/number-of-equivalent-domino-pairs/description/

풀이

처음 푼 방식이다. 어렵지 않게 풀 수 있었으며 시작 복잡도는 O(n) 이다. 지금 생각해보면 굳이 arr.sort()를 통해 정렬한뒤 join으로 새로운 배열을 생성한 뒤 다시 연산을 할게 아니고 처음 arr를 순회하면서 모든 연산을 처리할 수 있었다. 이런 개선점들이 보여 한번 더 정리를 했다. (하단 2번째 코드 확인)

/**
 * @param {number[][]} dominoes
 * @return {number}
 */
var numEquivDominoPairs = function (dominoes) {
    dominoes = dominoes.map((arr) => {
        const key = arr.sort((prev, next) => prev - next).join("");
        return key;
    });

    let map = new Map();

    for (const key of dominoes) {
        map.set(key, (map.get(key) || 0) + 1);
    }

    let pairs = 0;
    map.values().forEach((n) => {
        pairs += n * (n - 1) * 0.5;
    });

    return pairs;
};

두번째로 푼 방식이다. 연산이 더욱 간결해졌다.

/**
 * @param {number[][]} dominoes
 * @return {number}
 */
var numEquivDominoPairs = function (dominoes) {
    let map = new Map();
    let pairs = 0;
    for (const [n1, n2] of dominoes) {
        let key = ''+Math.max(n1, n2) + Math.min(n1, n2);
        map.set(key, (map.get(key) || 0) + 1);
        pairs += map.get(key) - 1;
    }

    return pairs;
};