JavaScript 배열에서 키별 최대값 추출하기

문제 상황

다음과 같이 객체로 구성된 배열이 있다고 가정해 보겠습니다.

const arr = [
    {a:1, b:"apples"},
    {a:3, b:"apples"},
    {a:4, b:"apples"},
    {a:1, b:"bananas"},
    {a:3, b:"bananas"},
    {a:5, b:"bananas"},
    {a:6, b:"bananas"},
    {a:3, b:"oranges"},
    {a:5, b:"oranges"},
    {a:6, b:"oranges"},
    {a:10, b:"oranges"}
];

여기서 요구되는 것은 이러한 배열을 입력받아 새로운 객체 배열을 반환하는 JavaScript 함수를 작성하는 것입니다.

반환되는 배열은 'b' 속성의 고유한 값마다 하나의 객체를 포함해야 하며, 각 객체는 동일한 'b' 값을 가진 항목들 중 'a' 속성 값이 가장 큰 것이어야 합니다.

해결 방법

이 문제는 임시 객체를 맵(map)처럼 활용하면 효율적으로 해결할 수 있습니다. 배열을 한 번만 순회하면서 각 'b' 값이 처음 등장하면 결과 배열에 추가하고, 이미 존재한다면 기존 객체의 'a' 값과 비교하여 더 큰 값으로 교체하는 방식입니다.

const arr = [
    {a:1, b:"apples"},
    {a:3, b:"apples"},
    {a:4, b:"apples"},
    {a:1, b:"bananas"},
    {a:3, b:"bananas"},
    {a:5, b:"bananas"},
    {a:6, b:"bananas"},
    {a:3, b:"oranges"},
    {a:5, b:"oranges"},
    {a:6, b:"oranges"},
    {a:10, b:"oranges"}
];

const pickHighest = arr => {
    const res = [], map = {};

    arr.forEach(el => {
        if (!(el['b'] in map)) {
            map[el['b']] = res.push(el) - 1;
            return;
        };
        if(res[map[el['b']]]['a'] < el['a']){
            res[map[el['b']]] = el;
        };
    });
    return res;
};
console.log(pickHighest(arr));

동작 원리

• res: 최종 결과를 담는 배열입니다.
• map: 각 'b' 값이 결과 배열의 몇 번째 인덱스에 저장되었는지 추적합니다.
• 새로운 'b' 값이 등장하면 해당 객체를 결과 배열에 추가하고 인덱스를 기록합니다.
• 이미 존재하는 'b' 값이라면 기존 객체의 'a' 값과 비교하여 더 큰 경우에만 교체합니다.

실행 결과

콘솔에 출력되는 결과는 다음과 같습니다.

[
    { a: 4, b: 'apples' },
    { a: 6, b: 'bananas' },
    { a: 10, b: 'oranges' }
]

이처럼 시간 복잡도 O(n)으로 배열을 단 한 번 순회하여 각 키별 최대값을 가지는 객체만 깔끔하게 추출할 수 있습니다.