[백준 알고리즘][자바] 1181번 : 단어 정렬

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

1181번: 단어 정렬

이번 문제는 단어를 정렬하는 문제입니다.

첫 번째 기준은 길이가 짧은 것 두 번째 기준은 길이가 같으면 사전 순으로 정렬하는 것입니다.

좌표 정렬하기에서는 퀵정렬을 수정하여 알고리즘을 직접 만들어서 해결했는데

이번 문제는 쉽지 않네요.

일단 Array.sort()를 사용하여 문제를 해결하고 직접 알고리즘을 만드는 거는 고민을 해봐야겠습니다.

 

 

풀이

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.Comparator;

public class Main {

	public static void main(String[] args) throws NumberFormatException, IOException {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		
		int N = Integer.parseInt(br.readLine()); // 단어의 개수
		String[] str_array = new String[N];
		 
		for (int i = 0; i < N; i++) {
			str_array[i] = br.readLine();
		}
		
		Arrays.sort(str_array, new Comparator<String>() {
			public int compare(String s1, String s2) {
				
				if (s1.length() == s2.length()) { // 단어 길이가 같을 경우 사전 순
					return s1.compareTo(s2);
				} else { // 길이가 다른 경우
					return s1.length() - s2.length();
				}
			}
		});
		
		for (int i = 0; i < N; i++) {
			if(i > 0) {
				if (!str_array[i].equals(str_array[i - 1])) {
					bw.write(str_array[i] + "\n");
				}
			}else {
				bw.write(str_array[i] + "\n");
			}
		}
	
		br.close();
		bw.flush();
		bw.close();
	}	
	
}