백준 1181 단어정렬 With Java

 

 

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

1181번: 단어 정렬

 

 

import java.util.Scanner;
import java.util.Arrays;
import java.util.Comparator;
 
 
public class Main {
    public static void main(String[] args) {
    
        Scanner in = new Scanner(System.in);
 
        int N = in.nextInt();
        String[] arr = new String[N];
 
        in.nextLine();    
        for (int i = 0; i < N; i++) {
            arr[i] = in.nextLine();
        }
        
        Arrays.sort(arr, 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();
                }
            }
        });
 
 
        System.out.println(arr[0]);
        
        for (int i = 1; i < N; i++) {
 
                        if (!arr[i].equals(arr[i - 1])) {
                System.out.println(arr[i]);
            }
        }
    }
 
}

 

먼저 스캐너를 선언을 해준다. 그 다음 스캐너를 통해서 사용자한테 단어의 갯수를 입력을 받는다. 그 다음 배열을 만들어 주고 N만큼 for문을 통해서 배열에 그 값을 넣어준다. in.nextLine();을 쓴 이유는 이를 쓰지 않고 바로 for문으로 들어가면 개행문자가 arr[0]에 들어가기 때문이다. 그 다음 new Comparator<String>()로 Arrays.sort의 Comparator를 재정의 해준다.

 

compare 메소드는 기본적으로 리턴형이 int형이다. 3가지의 리턴 값이 있는데 이 리턴 값에 의해 위치를 바꿀지 말지, 바꾼다면 어디로 바꿀지 결정한다. 3가지 리턴값은 1. 양수, 2. 0, 3. 음수이다. 

 

양수일 경우 두 위치는 바뀌고 0과 음수일 경우이면 바뀌지 않는다. 예를 들어 배열에 {3,4,5}라는 값이 있으면

 

3 - 4를 진행을 한다. 음수이기 때문에 아무일도 일어나지 않는다. 만약 {4,3,5}였다면? 4 - 3은 양수이기 때문에 {3,4,5}로 바뀌었을 것이다. 

 

그럼 그 다음 단어 길이가 같으면 s1.compareTo(s2)를 통해서 단어를 사전순으로 정렬을 한다.(문제의 조건)

 

그렇지 않으면 단어의 길이가 다르다면 두 단어의 길이를 빼주어서 비교를 해주어. 0인지, 양수인지, 음수인지 판별을 해준다