[백준 1181] 알고리즘 15일차 : 단어 정렬

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

1181번: 단어 정렬

[정렬] Java

오랜만에 자바로 푸니까 다 까먹음 진짜

반성하면서 앞으로 자바로 풀겠습니다..

 

문제가 딱 생긴게 자바로 풀면 딱이라 시도해봤는데

그건 맞는데 암튼

 

1. 길이로 정렬 -> Collections.sort사용 comparator써서 a.length()-b.length() 반환

2. 길이가 같으면 사전순으로 정렬 -> Collections.sort comparator에서 a.compareto(b) 사용

3. 중복없이 출력 -> HashSet 사용 (처음 써봄)

 

소스코드

 

import java.util.*;

public class CharSotring {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = Integer.parseInt(in.nextLine());

        //중복없이 입력받음
        HashSet<String> strSet = new HashSet<>();
        String str;
        for(int i = 0 ; i < n ; i++){
            str = in.nextLine();
            strSet.add(str);
        }
        //정렬 1.길이 2. 길이같으면 사전순
        ArrayList<String> arrList = new ArrayList<>(strSet);
        Collections.sort(arrList, (a,b)->{if(a.length() == b.length()) return a.compareTo(b); else return a.length()-b.length();});
        //출력
        for(String s : arrList)
            System.out.println(s);
    }
}