알고리즘 단계별로 풀어보기 : BOJ-1181(단어정렬)

문제(출처)

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로

입력

첫째 줄에 단어의 개수 N이 주어진다. (1≤N≤20,000) 

둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 

주어지는 문자열의 길이는 50을 넘지 않는다.

출력

조건에 따라 정렬하여 단어들을 출력한다. 

단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

예제 입력

13

but

i

wont

hesitate

no

more

no

more

it

cannot

wait

im

yours

예제 출력

i

im

it

no

but

more

wait

wont

yours

cannot

hesitate

내 풀이

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.util.Arrays;

import java.util.Comparator;

import java.util.HashSet;

public class Main{

    

    public static void main(String args[]) {

        String[] str = testCase1();

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

        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        

        

        HashSet<String> set = new HashSet<>();

        

        for(int i = 0; i < str.length; i++) {

            set.add(str[i]);

        }

        

        String[] temp = new String[set.size()];

        set.toArray(temp);

        

        Arrays.sort(temp, new Comparator<String>() {

            @Override

            public int compare(String o1, String o2) {

                int length1 = o1.length();

                int length2 = o2.length();

                if(length1 == length2) {

                    return o1.compareTo(o2);

                }else {

                    return length1 - length2;

                }

            }

        });

        

        

        

        

        for(int i=0; i < temp.length; i++) {

            System.out.println(temp[i]);

        }

        

    }

    

    

    public static String[] testCase1() {

        String[] str =

{ "but", "i", "wont", "hesitate", "no", "more", "no", "more", "it", "cannot", "wait", "im",

                "yours" };

        return str;

    }

}

내 풀이 해석

해쉬셋으로 중복을 제거하고 배열에 담에 정렬 시 2가지 조건으로 정렬합니다.

아쉬운 점

해쉬맵을 정렬 하는 방법을 생각하느라 시간이 걸렸습니다.

위 코드가 틀렸다는데 틀린 이유를 찾지 못하여 참고했습니다.

아스키 코드로 변환하여 비교하고 정렬했는데 아스키 코드로 정렬 하려고 했던것 자체가 문제를 이해를 잘못했던거였습니다.