[BaekJoon] 1181번 단어 정렬 문제풀이 (Success)

1181번: 단어 정렬

 

☕Language: Java

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        int N = Integer.parseInt(br.readLine());
 
        Map<String, Integer> map = new HashMap<>();
 
        for (int i = 0; i < N; i++) {
            String s = br.readLine();
            map.put(s, s.length());
        }
 
        Object[] mapKeys = map.keySet().toArray();
        Arrays.sort(mapKeys);
 
        List<Object> keys = new ArrayList<>();
        for (Object key : mapKeys) {
            keys.add(key);
        }
        Collections.sort(keys, ((k1, k2) -> (map.get(k1).compareTo(map.get(k2)))));
 
        for (Object key : keys) {
            bw.write(key + "\n");
        }
 
        bw.flush();
        bw.close();
    }
 
}
 

🤔 해설

1. Map<String, Integer> map = new HashMap<>();

    - Key: 이름, Value: 이름 길이

 

2. Arrays.sort(mapKeys);

    - 이름 길이에 따라 정렬 전에 이름(key)을 기준으로 정렬 수행

    - toArray(): 기본 return type이 Object[]

    - String 배열 선언 방법: String[] mapKeys =  map.keySet().toArray(new String[0]);

      ⭐ String[] mapKeys =  (String[]) map.keySet().toArray(); 시, ClassCastException 발생

        - [해결 방법] java.lang.ClassCastException

 

3. List<Object> keys = new ArrayList<>();

    - Collections.sort를 사용하기 위해, keys에 해당하는 List 선언

 

4. Collections.sort(keys, ((k1, k2) -> (map.get(k1).compareTo(map.get(k2)))));

    - value(map.get())에 따른 오름차순으로 정렬

    - 내림차순 정렬: map.get(k2).compareTo(map.get(k1))

    - 오름차순 정렬: map.get(k1).compareTo(map.get(k2))

    - A.compareTo(B)

        - -1: A < B ▶ A(앞)가 B(뒤)의 뒤에 위치

        - 0: A == B ▶위치 변화 X

        - 1: A > B ▶ A(앞)가 B(뒤)의 앞에 위치

 

5. map 자료형은 중복을 허용하지 않으므로, 중복 조건은 따로 신경쓰지 않아도 됨

 

😮 찾아본 풀이

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 IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int N = Integer.parseInt(br.readLine());
 
        String[] sArr = new String[N];
 
        for (int i = 0; i < N; i++) {
            sArr[i] = br.readLine();
        }
 
        Arrays.sort(sArr, 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();
                }
            }
 
        });
 
        StringBuffer sb = new StringBuffer();
 
        sb.append(sArr[0] + "\n");
 
        for (int i = 1; i < N; i++) {
            if (!sArr[i].equals(sArr[i - 1])) {
                sb.append(sArr[i] + "\n");
            }
        }
 
        bw.write(sb + "");
 
        bw.flush();
        bw.close();
    }
 
}
 

1. Arrays.sort(sArr, new Comparator<String>() {...}

    - 입력값을 String 배열로 저장해서 정렬

 

2. public int compare(String s1, String s2) {...}

    - 길이가 같을 경우, 오름차순 정렬(사전식 정렬)

    - 길이가 다를 경우, 길이가 짧은 문자열부터 정렬

  ▶ Arrays.sort() 결과는 단어의 길이가 짧은 순서 + 단어의 길이가 같은 경우에는 사전순으로 출력됨

 

3. if (!sArr[i].equals(sArr[i - 1])) {...}

    - 중복 제거

 

 

 

🔗 소스 코드
HJ0216/TIL/BOJ

 

📚 참고 자료

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