JavaScript로 배열의 고유 항목 개수를 세는 함수 만들기

배열에 포함된 모든 고유(unique) 항목의 개수를 세는 JavaScript 함수를 작성해 보겠습니다. 이 함수는 배열을 순회하면서 각 요소가 몇 번 등장하는지 계산한 뒤, 그 결과를 객체 형태로 반환합니다.

문제 정의

예를 들어 다음과 같은 배열이 있다고 가정해 보겠습니다.

const arr = ["hi", "hello", "hi"];

이 배열에는 "hi"가 2번, "hello"가 1번 등장하므로, 함수는 { hi: 2, hello: 1 }과 같은 객체를 반환해야 합니다.

구현 방법

빈 객체를 하나 준비한 뒤, 반복문으로 배열을 순회하면서 각 요소를 키(key)로 사용해 등장 횟수를 누적하는 방식으로 구현할 수 있습니다.

const arr = ["hi", "hello", "hi"];
const countUnique = arr => {
    const counts = {};
    for (var i = 0; i < arr.length; i++) {
        counts[arr[i]] = 1 + (counts[arr[i]] || 0);
    };
    return counts;
};
console.log(countUnique(arr));

코드 설명

핵심 로직은 counts[arr[i]] = 1 + (counts[arr[i]] || 0) 부분입니다. 어떤 요소가 처음 등장하면 counts 객체에 아직 해당 키가 없어 undefined가 되고, 논리 OR(||) 연산자에 의해 0으로 대체됩니다. 따라서 첫 등장 시 1이 저장되고, 이후에는 기존 값에 1씩 더해지며 등장 횟수가 누적됩니다.

출력 결과

위 코드를 실행하면 콘솔에 다음과 같은 결과가 출력됩니다.

{ hi: 2, hello: 1 }

대안: reduce 메서드 활용

같은 기능은 Array.prototype.reduce() 메서드를 사용하면 더 간결하게 구현할 수 있습니다.

const countUnique = arr =>
  arr.reduce((acc, item) => {
    acc[item] = (acc[item] || 0) + 1;
    return acc;
  }, {});

console.log(countUnique(["hi", "hello", "hi"])); // { hi: 2, hello: 1 }

두 방식 모두 시간 복잡도는 O(n)으로 배열의 길이에 비례하며, 문자열뿐 아니라 숫자 등 다른 원시 타입의 요소에도 동일하게 적용할 수 있습니다.