[백준(BOJ)] 1181번 : 단어 정렬 - JAVA[자바]

www.acmicpc.net/problem/1181

1181번: 단어 정렬

 

---

 

 

⊙ 문제

⊙ 입력

⊙ 출력

⊙ 예제 입출력

⊙ 알고리즘 분류

• 문자열
• 정렬

 

---

 

⊙ 문제 접근 과정

 

배열이 특정한 규칙( 1. 길이가 짧은 것부터, 2. 길이가 같으면 사전 순으로)을 통해 정렬하라는 걸 보고

함수를 어떻게 만들어야할지 고민을 했다. 생각이 끝나고 함수를 구현하고 있었는데 코드가 말도 안 되게

길어지는 것을 보고 아니라고 생각했다.

 

그러다 Arrays.sort의 Comparator를 알게 되었다.

 

Comparator를 사용하자 코드 길이가 많이 짧아졌다.

 

---

 

⊙ Comparator

 

• Comparator란?

       Comparator는 객체 간 특정한 규칙이 필요할 때 사용한다.

 

---

 

⊙ 문제 풀이

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        String[] str = new String[N];

        for (int i = 0; i < N ; i++) {
            str[i]=in.next();
        }

        Arrays.sort(str, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if (o1.length() == o2.length()) {
                    return o1.compareTo(o2);
                }
                else {
                    return o1.length() - o2.length();
                }
            }
        });

        System.out.println(str[0]);
        for(int i =1 ; i < N; i++) {
            if (!str[i].equals(str[i-1])) {
                System.out.println(str[i]);
            }
        }
    }
}

 

 

 

---

⊙ 결과

 

 

---

⊙ 마무리

 

 

배열에 특정한 정렬이나 일반적인 정렬이 필요할 때, 자바에서는 Comparator나 Comparable를 활용하자.