[백준BOJ] 1181번 단어 정렬.java

 

백준 저지에서 단어 정렬을 자바를 통해 풀어 보았다. 

 

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

1181번: 단어 정렬

1181번 단어 정렬.java

GitHub - tomy9729/TargetIsPlatinum5: 👯목표는 플레티넘5👯

 

1181번 단어 정렬

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

설명

여러 개의 주어진 단어들에 대해서 길이가 짧은 것부터, 길이가 같은면 사전 순으로 정렬해야한다. 이런 조건이 달린 정렬을 할 때 자바에서는 Comparator라는 interface를 사용할 수 있다. Comparator에서 compare를 오버라이딩하여 정렬의 새로운 기준을 정의할 수 있다. 

 

만약 두 단어의 길이가 같다면 사전순으로 반환하고, 그렇지 않다면 길이순으로 반환한다.

 

여기서 사전순은 String의 compareto를 통해 가능하다.

코드

//1181번 단어 정렬.java
package week2;

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

class Word{
	public String w;
	public int len;
	
	Word(String w, int len){
		this.w = w;
		this.len = len;
	}
}

public class 단어_정렬_1181 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		String word[] = new String[N];
		
		for(int i=0;i<N;i++) {
			word[i]=sc.next();
		}	
		
		Arrays.sort(word, new Comparator<String>() {
			@Override
			public int compare(String w1, String w2) {
				int l1 = w1.length();
				int l2 = w2.length();
				if(l1==l2)
					return w1.compareTo(w2);
				return l1<l2 ? -1 : 1;
			}
		});
		
		System.out.println(word[0]);
		for(int i=1;i<N;i++) {
			if(!word[i].equals(word[i-1]))System.out.println(word[i]);
		}
		
	}
}