BOJ 1181 - 단어 정렬

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

1181번: 단어 정렬

 

---

 

직전 문제와 거의 비슷한 문제이지만 문제를 살펴보면 단어의 중복 제거라는 고민해야 할 부분이 있다.

이를 위해 자바 8 이후부터 권장되는 Stream을 사용해 중복을 제거해 보자.

 

Stream에서 제공되는 API를 사용하기 위해서는 먼저 String[] 배열을 Stream 객체로 변환시켜 주어야 한다.

그리고 Stream API를 사용하여 중복을 제거하고 다시 String[] 배열로 변환시켜 주면 된다.

 

자세한 코드는 다음과 같다.

 

 

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        int number = Integer.parseInt(br.readLine());
        String[] words = new String[number];

        for (int i = 0; i < number; i++) {
            words[i] = br.readLine();
        }

        words = Arrays.stream(words).distinct().toArray(String[]::new);

        Arrays.sort(words, (e1, e2) -> {
            if (e1.length() == e2.length()) {
                for (int i = 0; i < e1.length(); i++) {
                    if (e1.charAt(i) != e2.charAt(i)) {
                        return e1.charAt(i) - e2.charAt(i);
                    }
                }
            }

            return e1.length() - e2.length();
        });

        for (int i = 0; i < words.length; i++) {
            sb.append(words[i]).append("\n");
        }

        System.out.println(sb);
        br.close();
    }
}