백준 문제 풀이

[자바] 백준 1181 단어 정렬

소힌 2022. 3. 27. 20:50
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;

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

길이가 짧은 것부터
길이가 같으면 사전 순으로
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 
둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 
주어지는 문자열의 길이는 50을 넘지 않는다.

출력
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.*/
public class bj1181 {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		System.out.println("입력 문자열 개수를 입력하세요 ");
		int num = scan.nextInt();

		String[] list = new String[num];

		for (int i = 0; i < num; i++) {
			list[i] = scan.next();
		}

		Arrays.sort(list, 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 < list.length; i++) {
        if(i!=0 && list[i].equals(list[i-1)) {
			System.out.println(list[i]);}
		}

	}
}

ㅠㅠ 오랜만에 했더니 넘 어렵다.... 

여기서 정리할 것은 Arrays.sort (정렬대상 , Comparator<>()) 

그리고 사전 비교 compareTo 

길이 비교 값을 리턴하면 길이 순대로 정렬되는것 

 

알면 편한데 모르면 정말 어려울 것 같다