[코딩 테스트/백준 알고리즘] 1181번 : 단어 정렬 (Java 풀이)

문제

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

1181번: 단어 정렬

---

문제 설명

---

문제 풀이

 문제 조건에 맞게 Comparator를 구현해주면 된다.

 

2022.08.01 - [✏️Java 공부/기타 등등] - [Java 공부/기타] 객체 정렬하기 (Comparable & Comparator)

[Java 공부/기타] 객체 정렬하기 (Comparable & Comparator)

이 글을 참고하면 도움이 될 것이다.

---

코드

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

class Main {
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        HashSet<String> set = new HashSet<>();


        while(N-- > 0) {
            set.add(br.readLine());
        }
        String[] arr = new String[set.size()];
        set.toArray(arr);

        Arrays.sort(arr, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                if (s1.length() == s2.length()) {	//단어 길이가 같은 경우
                    return s1.compareTo(s2);	//단어 순서 비교
                }
                else {	//단어 길이가 다른 경우
                    return s1.length() - s2.length();	//단어 길이 비교
                }
            }
        });
        for(String str : arr) {
            System.out.println(str);
        }
    }
}