백준 알고리즘
[Silver V] 단어 정렬 - 1181
성능 요약
메모리: 15368 KB, 시간: 832 ms
분류
문자열, 정렬
문제 설명
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
단, 중복된 단어는 하나만 남기고 제거해야 한다.
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
출력
조건에 따라 정렬하여 단어들을 출력한다.
Code
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
use std::io::{self, BufRead};
use std::cmp::Ordering;
fn cmp(a: &String, b: &String) -> Ordering {
// 1. 길이가 같다면, 사전순으로
if a.len() == b.len() {
return a.cmp(b);
}
// 2. 길이가 다르다면, 짧은 순으로
else {
return a.len().cmp(&b.len());
}
}
fn main() {
let stdin = io::stdin();
let mut word: Vec<String> = vec![];
// N 입력받기
let mut input = String::new();
stdin.lock().read_line(&mut input).unwrap();
let n: usize = input.trim().parse().unwrap();
// word 배열 입력받기
for _ in 0..n {
let mut input = String::new();
stdin.lock().read_line(&mut input).unwrap();
word.push(input.trim().to_string());
}
// word 정렬하기
word.sort_by(cmp);
// 중복 제거 후 출력
let mut prev_word = "";
for w in word.iter() {
if w != prev_word {
println!("{}", w);
prev_word = w;
}
}
}