[백준] 1181번 : 단어 정렬

1181번: 단어 정렬

---

문제

---

소스 코드

---

풀이

ⓐ 테스트 케이스 개수를 받을 변수 N을 생성한다.

BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

int N = Integer.parseInt(br.readLine());

ⓑ 문자열을 받을 배열 arr과 for문을 통해 문자열을 넣어준다.

String[] arr = new String[N];

for (int i = 0 ; i < N; i++) {
    arr[i] = br.readLine();
}

ⓒ Arrays.sort()를 통해 정렬을 한다. Comparator를 람다식으로 구현해 넣어준다.

Arrays.sort(arr,(s1, s2) -> {
    if (s1.length() == s2.length()) {
        return s1.compareTo(s2);
    }
    else return s1.length() - s2.length();
} );

위 코드는 arr[0].length() == arr[1].length() 일 경우 사전순(compareTo())으로 정렬하고,

그 외의 상황에서는 s1.length() - s2.length()를 한다는 뜻이다.

Comparator은 기본적으로 반환타입이 정수형인데,

음수와 0일 경우에는 자리를 바꾸지 않고, 양수일 경우에는 정렬 알고리즘을 통해 자리를 바꾼다.

 

ⓓ 배열 arr의 [0]번 인덱스만 따로 먼저 출력한다.

System.out.println(arr[0]);

 

ⓔ for문을 통해 출력을 하지만 만약 arr[i]번 인덱스 값과 arr[i-1] 인덱스 값이 같지 않을경우에만 출력하게 한다.

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

---

참조

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