[JAVA]백준 - 1181.단어 정렬(compare 오버라이딩)

www.acmicpc.net/problem/1181

1181번: 단어 정렬

 

풀이 > 

중복값을 제거하기위해 Set에 값을 넣어주고 ArrayList로 변환
list.sort의 compare을 오버라이딩해 길이로 비교하고, 길이가 같다면
시작 문자로 비교하여 정렬하기

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

public class Baekjoon1181 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int count = Integer.parseInt(br.readLine());

		// 중복값 제거
        HashSet<String> set = new HashSet<>();

        for (int i = 0; i < count; i++)
            set.add(br.readLine());

        ArrayList<String> list = new ArrayList<>(set);

        list.sort((s1, s2) -> {
            if(s1.length() > s2.length())
                return 1;
            else if(s1.length() < s2.length())
                return -1;
            else
                return s1.compareTo(s2);
        });

        for (String s : list)
            bw.append(s).append("\n");

        bw.flush();
        br.close();
        bw.close();
    }
}