[백준]JAVA - 1181번: 단어 정렬

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

1181번: 단어 정렬

풀이

이번 문제는 기본적인 오름차순 정렬로는 해결할 수 없기 때문에 Arrays.sort() 메서드에서 Comparator를 구현해야한다.

 

https://sookr5416.tistory.com/151

[백준]JAVA - 1931번: 회의실배정

정렬에 대해서 궁금하다면 위 포스팅을 읽어보는걸 추천한다.

 

기본적인 오름차순 정렬이 아닌 특정한 규칙에 의한 정렬을 하고 싶은 경우에는 Comparator를 사용해야한다. 

간단하게 설명하자면 Arrays.sort(정렬할 배열, new Comparator<정렬할 배열의 타입>() {}으로 시작해서 내부에 Compare 메서드를 오버라이딩해주는 것이다. 이 문제의 조건으로 오버라이딩하면 아래와 같다.

Arrays.sort(str, new Comparator<String>() {
    public int compare(String s1, String s2) {
        if (s1.length() == s2.length()) {
            return s1.compareTo(s2);
        }
				
        return s1.length() - s2.length();
    }
});

길이가 같다면 단어 사전순 정렬을 하라고 했으니까 이건 간단하게 compareTo() 메서드로 해주면 된다. 

 

<최종코드>

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

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());
        String[] str = new String[N];
		
        for (int i=0; i<N; i++) {
            str[i] = br.readLine();
        }
		
        // 정렬 재정의
        Arrays.sort(str, new Comparator<String>() {
            public int compare(String s1, String s2) {
                if (s1.length() == s2.length()) {
                    return s1.compareTo(s2);
                }
				
                return s1.length() - s2.length();
            }
        });
		
		
        String tmp = "";
        for (String s : str) {
            if (s.equals(tmp)) continue;
			
            System.out.println(s);
            tmp = s;
        }
    }
}