백준 BAEKJOON | 1181 - 단어 정렬 JavaScript

github 문제 풀이 코드

📌 문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로
3. 단, 중복된 단어는 하나만 남기고 제거해야 한다.

• 첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000)
• 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다.
• 주어지는 문자열의 길이는 50을 넘지 않는다.

입력	출력
13 but i wont hesitate no more no more it cannot wait im yours	i im it no but more wait wont yours cannot hesitate

✨ 해결 방법

우선 배열 input을 알파벳 string 각각의 배열로 가공합니다.
갯수를 나타내는 가장 첫 번째 요소를 삭제하고, set을 사용하여 중복을 제거합니다.
이를 다시 정렬하기 위해 배열로 만들고, 두 가지 조건으로 정렬합니다.
1. 길이가 짧은 순으로 정렬
2. 길이가 같을 경우 -> 알파벳 순으로 정렬

💻 solution

const input = require('fs')
    .readFileSync("/dev/stdin")
    .toString().trim().split('\n');

const uniqueWords = new Set([...input].slice(1)); // 중복 제거
const sortedWords = [...uniqueWords].sort((a, b) => a.length - b.length || a.localeCompare(b)))); // 재정렬

console.log(arr.join('\n'))

||의 경우 앞의 것을 따지고, 해당하지 않을 경우 넘어오기 때문에 길이가 같은 경우에만 비교할 수 있습니다.

❓ sort() 알파벳 순으로 정렬하는데 localeCompare을 사용한 이유가 뭔가요?
물론, 대부분의 경우엔 .sort().sort((a, b) => a.length - b.length) 와 위의 식이 동일하게 동작합니다. 하지만 sort는 항상 ASCII 코드 순서를 기준으로 하기 때문에, 가령 아래와 같은 경우에선 예상과 다르게 동작할 수 있습니다.

const words = ["éclair", "elephant", "apple", "Étoile", "banana"]; (프랑스어 기준)

sort() 기본 정렬 (ASCII)	localeCompare("fr") 프랑스어 기준
Étoile	apple
apple	banana
banana	elephant
elephant	éclair
éclair	Étoile

---

💻 solution2

다른 방법으로, sort로 길이로만 정렬한 후stack을 사용하여 알파벳 순으로 정렬할 수 있습니다.

const input = require('fs')
    .readFileSync(process.platform === "linux" ? "/dev/stdin" : "./input.txt")
    .toString().trim().split('\n');

const uniqueWords = new Set([...input].slice(1)); // 중복 제거
const sortedByLength = [...uniqueWords].sort((a, b) => a.length - b.length); // 길이순 정렬

let stack: string[] = [];
let result: string[] = [];

// ✅ stack을 활용한 정렬
for (let word of sortedByLength) {
    if (!stack.length || stack[stack.length - 1].length === word.length) {
        stack.push(word);
    } else {
        // 현재 단어와 길이가 다르면 stack 내부를 정렬하여 결과에 추가
        stack.sort((a, b) => a.localeCompare(b));
        result.push(...stack);
        stack = [word]; // 새로운 길이의 단어 추가
    }
}

// ✅ 마지막 남은 stack 정렬 후 result에 추가
if (stack.length) {
    stack.sort((a, b) => a.localeCompare(b));
    result.push(...stack);
}

// ✅ 최종 출력
console.log(result.join("\n"));