[백준OJ] 1181번 단어 정렬

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

1181번: 단어 정렬

---

 

풀이

현재까지 나온 문자열인지 확인을하는 Map을 이용해서 나온적이 없는 문자열이면 벡터에 add한 뒤, 정렬기준을 문자열의 길이가 작은순, 같다면 사전적으로 먼저오는순으로 정렬을 한다.

 

코드

import java.util.*;

public class Main {
    static Vector<String> v=new Vector<>();
    static Map<String,Boolean> check=new HashMap<>();
    static int n;
    public static void main(String[] args){
        Scanner scanner=new Scanner(System.in);
        n=scanner.nextInt();

        for(int i=0;i<n;i++){
            String temp=scanner.next();
            if(check.get(temp)==null){
                check.put(temp,true);
                v.add(temp);
            }
        }
        Collections.sort(v, 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();
                }
            }
        });
        for(int i=0;i<v.size();i++){
            String str=v.get(i);
            System.out.println(str);
        }
    }
}