[백준] 1181 - 단어 정렬 (JAVA/ 자바)

1181번 문제(1)

1181번 문제(2)

 

처음에 푼 방식이 맞았다고 뜨긴 했는데 다시 넣어보니 시간초과 이슈가 발생했다. 그래서 BufferedReader를 사용하여 

최대한 시간을 줄여보았다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

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());
    List<String> list = new ArrayList<>();

    for(int i=0; i<N; i++){
        list.add(br.readLine());
    }
    //1.단어 중복 제거
    list = list.stream().distinct().collect(Collectors.toList());
    String[] arr = list.toArray(new String[list.size()]);  
    //2.사전순으로 정렬
    Arrays.sort(arr);   
    //3.길이 비교
    for(int i=0; i<arr.length; i++){
        Arrays.sort(arr, (String s1, String s2) -> s1.length() - s2.length());      
    }
    for(int i=0; i<arr.length; i++){
        System.out.println(arr[i]);
    }
br.close();
}
}

 

Comparator 익명 클래스를 사용한 방법

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

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());
    List<String> list = new ArrayList<>();

    for(int i=0; i<N; i++){
        list.add(br.readLine());
    }
    //1.단어 중복 제거
    list = list.stream().distinct().collect(Collectors.toList());
    String[] arr = list.toArray(new String[list.size()]);
    
    //2. 정렬하기
    Arrays.sort(arr, new Comparator<String>() {
        @Override
        public int compare(String s1, String s2){
            if(s1.length() == s2.length()) {
                return s1.compareTo(s2);
            }else{
                return s1.length() - s2.length();
            }
        }    
    });
    for(int i=0; i<arr.length; i++){
        System.out.println(arr[i]);
    }
br.close();
}
}

 

[참고한 블로그]

중복 제거

https://seungwontech.tistory.com/23

[Java] distinct()

ArrayList를 배열로 변환

https://mommoo.tistory.com/32

JAVA - ArrayList에서 배열로, 배열에서 ArrayList로

문자열 길이 기준 정렬 방법

https://sdardew-valley.tistory.com/59

[Java] 문자열 배열을 길이를 기준으로 정렬하기

 

이렇게 보니까 문제 해결 우선순위만 내가 결정하고 코드 짜기는 한 게 없는 거 같다...