[백준 - 1181][Java] 단어 정렬

이 포스팅은 자바 8버전으로 풀이가 됐습니다.

 

문제 출처

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

1181번: 단어 정렬

---

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

• 길이가 짧은 것부터
• 길이가 같으면 사전 순으로

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

출력

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.io.IOException;
// 1181번
public class Main {
    public static void main(String[] args) {
        try {
            
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            
            int arr_num = Integer.parseInt(br.readLine());
            
            String text[] = new String [arr_num];
            
            for(int i=0;i<arr_num;i++) {
                String data = br.readLine();
                text[i] = data;
            }
            Arrays.sort(text);
            Arrays.sort(text, new Comparator<String>() {
                @Override
                public int compare(String test1, String test2) {
                    return test1.length() - test2.length();
                }
            });
            
            text = Arrays.stream(text).distinct().toArray(String[]::new); // 중복 제거
            for(int k=0;k<text.length;k++) {
                System.out.println(text[k]);
            }
        }catch(IOException e) {
            System.out.println(e.toString());
        }
    }
}

풀이

문제 자체가 엄청 어려운 수준의 문제는 아니었다.
처음에 13이라는 문자 배열의 크기 값을 입력받는다.
입력받은 값을 문자열 배열에 크기로 선언한다.

첫 반복문에서 모든 값들을 받는다.

모든 값을 오름차순으로 1차 정렬을 한다.

Arrays.sort에 제공되는 Comparator를 활용하여 문자열(String)의 길이 차이 값만큼 계산하여 배열의 값을 반환한다.


이후 중복 값을 제거하는 코드를 활용하여 중복 값을 제거한다. (1.8jdk 이상 가능)

정렬된 값들을 출력한다.

 

---

이해가 안가는 부분이나 오타, 문제가 있는 경우 댓글 부탁드리겠습니다!

다음 포스팅 때 뵙겠습니다!

 

 

(2022-08-01 간략 후기)

문제를 풀어내는 것에 초점을 맞춰서 문제를 풀어내긴 했지만 1등 분이 120초대 나왔는데 6.x배가 차이 날 정도로 효율성이 너무 좋지 못했기 때문에 다음에 기회가 된다면 다른 풀이 방법을 찾아서 더 빠른 코드를 구현하는 것을 목표로 해야겠습니다.