백준 1181번 단어 정렬(Java)

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

1181번: 단어 정렬

 

 

 

import java.util.*;
 
public class WordSort{
 
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        HashSet<String> set = new HashSet<>();
        int n = input.nextInt();
        for (int i = 0; i < n; ++i) {
            set.add(input.next());
        }
 
        ArrayList<String> list = new ArrayList<>(set);
        Collections.sort(list, new Comparator<String>() {
            @Override
            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);
    }
}
 

 

Comparator 인터페이스를 이용해서 정렬의 기준을 재정의 했다. 길이가 짧은 순서대로 정렬을 하고 길이가 같다면 다시 두개를 비교해서 사전순으로 정렬하는 것이다. 

 

 

https://devlog-wjdrbs96.tistory.com/68?category=830071

[JAVA]Array.sort 와 Collections.sort 의 차이

여기서 Comparator 과 Comparable 의 차이를 이해하고 문제를 풀어보자.