[알고리즘-백준(정렬)] 자바 1181번 단어 정렬 문제풀이. (Comparator을 이용하여 정렬 기준 변경)

문제

 

 

입력 및 출력 예제

 

문제 이해

 

문제에 바로 조건에 따라 정렬하는 프로그램을 작성하시오 라는 말이 있기 때문에 이 문제는 정렬 문제로 분류하여 푼다.

하지만 정렬 기준이 두 가지로 나뉜다.

 

1. 길이가 짧은 것 부터
2. 길이가 같으면 사전순으로
3. 중복 제거

 

지금까지 우리가 정렬을 할 때 int의 오름차순 혹은 내림차순 정렬을 했지만 이 문제는 특이하게 문자의 길이를 요구하므로 해당 요구조건을 만족시키기 위해서는 Comparator 인터페이스를 이용해서 해결해야 한다.

comparator

comparator의 알고리즘 문제풀이 사용법 을 통해서 사용법을 숙지하시고 오는것도 방법이에요 :)

문제 해결 접근

• 길이가 짧은 순으로 정렬 을 수행할 때는 str.length() 를 이용하여 수행
• 길이가 같으면 사전순 정렬 은 기본적으로 Arrays.sort(/String/)가 사전순 정렬을 지원하므로 같은 길이끼리 묶는것이 관건
• 중복 제거는 for 반복을 수행하며 배열의 끝 까지 현재 index의 값과 다음 index의 값이 같으면 출력 시에 출력하지 않고 넘어가기

소스코드

 

길이가 짧은 순으로 정렬

// 단어의 길이가 짧은순 정렬
        Arrays.sort(word, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(o1.length() > o2.length()) return 1;
                else if(o1.length() == o2.length()) return 0;
                else return -1;
            }
        });

두 객체의 비교를 위해 comparator의 compare 메서드를 재정의 한다.
여기서 주의해야할 점은 o1.length()로 단어의 길이로 비교를 했던 것이다.

 

길이가 같으면 사전순 정렬

for (int i = 0; i < n; i++) {
            int j;
            for (j = i+1; j < n; j++) {
                if(word[i].length() != word[j].length()) break;
            }
            Arrays.sort(word, i,j);
            i = j-1;
        }

 

중복 제거

// 중복 제거 후 출력
        System.out.println(word[0]);
        for (int i = 1; i < n; i++) {
            if(word[i-1].equals(word[i])) continue;

            System.out.println(word[i]);
        }

 

전체 소스코드

package algorithm.class04_Sort;

import java.util.*;

public class Prob06_WordSorting {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);

        int n = input.nextInt();

        String[] word = new String[n];

        for (int i = 0; i < word.length; i++) {
            word[i] = input.next();
            word[i] = word[i].toLowerCase();
        }



        // 단어의 길이가 짧은순 정렬
        Arrays.sort(word, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(o1.length() > o2.length()) return 1;
                else if(o1.length() == o2.length()) return 0;
                else return -1;
            }
        });

        // 단어의 길이가 같을 때 사전순 정렬
        for (int i = 0; i < n; i++) {
            int j;
            for (j = i+1; j < n; j++) {
                if(word[i].length() != word[j].length()) break;
            }
            Arrays.sort(word, i,j);
            i = j-1;
        }

        // 중복 제거 후 출력
        System.out.println(word[0]);
        for (int i = 1; i < n; i++) {
            if(word[i-1].equals(word[i])) continue;

            System.out.println(word[i]);
        }

    }
}