1181 - 단어 정렬

https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

문제 풀이

N 개의 단어를 받아서

 

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로 정렬하여 출력하는 문제

 

중복된 단어의 경우 한번만 출력한다는 조건이 하나 더 있으므로 입력받은 문자를 Set 자료구조에 담아서 중복을 제거한 뒤에 ArrayList로 옮겨서 정렬

 

package ex01;

import java.util.*;
import java.io.*;
 

 
class BOJ1181 {
    public static void main(String args[]) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        
        // 중복 제거를 위해 set으로 먼저 input
        HashSet<String> set = new HashSet<String>();
 
        for(int i=0; i<N; i++) 
            set.add(br.readLine());
        
        // List 변환
        ArrayList<String> list = new ArrayList<String>(set);
 
        // Comparator 클래스를 통하여 custom 정렬
        // 길이에 따라서 먼저 정렬하고 길이가 같으면 사전순으로 정렬
        Collections.sort(list, new Comparator<String>() {
            public int compare(String v1, String v2) {
                if(v1.length() > v2.length()) 
                    return 1;
                else if(v1.length() < v2.length()) 
                    return -1;
                else
                    return v1.compareTo(v2);
            }
        });
 
        for(String s : list) 
            System.out.println(s);
    }
}