치춘짱베리굿나이스

[백준] 2910 본문

빈도 정렬

문제

위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.

창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.

만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.

이렇게 정렬하는 방법을 빈도 정렬이라고 한다.

수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.

입력

첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)

둘째 줄에 메시지 수열이 주어진다.

출력

첫째 줄에 입력으로 주어진 수열을 빈도 정렬한 다음 출력한다.

풀이

const hack = () => {
  let [[n, c], arr] = require("fs")
    .readFileSync("/dev/stdin")
    .toString()
    .trim()
    .split("\n")
    .map((v) => v.split(" ").map(Number));
  let obj = {};
  for (let i = 0; i < n; i++) {
    if (!obj[arr[i]]) obj[arr[i]] = [i, 1];
    else obj[arr[i]][1]++;
  }
  console.log(
    Object.entries(obj)
      .sort((a, b) => {
        if (b[1][1] === a[1][1]) return a[1][0] - b[1][0];
        else return b[1][1] - a[1][1];
      })
      .map((v) => Array(v[1][1]).fill(v[0]).join(" "))
      .join(" ")
  );
};

hack();

반성회

  1. 각 숫자별 등장 빈도를 저장하기 위한 객체를 하나 만든다
  2. arr[i] (입력값의 각 숫자) 를 key로 사용하여, 만약 존재하지 않는 entry 일 경우 [i, 1]로 초기화하고 그 외의 경우 value의 1번째 인덱스를 1 증가시킨다 (i를 넣은 이유는 안정 정렬을 구현하기 위함)
  3. 객체를 배열로 만든다 (Object.entries(obj))
  4. 정렬하되, b[1][1]a[1][1] (등장 빈도수) 이 같을 경우 a[1][0]b[1][0]을 비교한다 (등장 순서, 앞에서 i를 저장한 부분이므로 작을 수록 먼저 등장한 수이다) 이를 통해 기존의 배열 순서가 유지되는 안정 정렬이 가능하다
  5. 등장 빈도만큼 수를 반복한다 (Array(v[1][1]).fill(v[0])) 객체로 만들면서 빈도수만 저장하고 정작 반복하여 등장한 수들은 사라졌기 때문에 다시 만들어주는 것
  6. 공백을 기준으로 출력한다

'Javascript + Typescript > 자바스크립트로 알고리즘풀기' 카테고리의 다른 글

[백준] 1620  (0) 2022.06.27
[백준] 2407  (0) 2022.06.26
[백준] 20492  (0) 2022.06.26
[백준] 2910  (0) 2022.06.25
[백준] 24860  (0) 2022.06.25
Comments