[알고리즘/백준] 1181 단어 정렬(자바)

문제

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

풀이 코드

• Stream API를 사용하여 중복제거, 정렬
  • distince() : 중복 제거
  • sorted() : 정렬
• StringBuilder사용을 통해 결과를 모아서 한번에 출력하여 실행 시간 단축

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        List<String> list = new ArrayList<>();

        for (int i = 0; i < n; i++) {
            list.add(br.readLine());
        }

        StringBuilder sb = new StringBuilder();

        list.stream().distinct()
                .sorted((s1, s2) -> {
                    if (s1.length() == s2.length()) {
                        return s1.compareTo(s2);
                    } else {
                        return s1.length() - s2.length();
                    }
                })
                .forEach(s -> sb.append(s).append("\n"));

        System.out.println(sb);
    }
}