백준 1181번. 단어 정렬 (Java)

문제 링크 : https://www.acmicpc.net/problem/1181

N 개의 단어를 받아서

1. 길이가 짧은 것부터

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

정렬하여 출력하는 문제입니다.

하지만 문제를 자세히 읽어보면 중복된 단어의 경우 한번만 출력한다는 조건이 하나 더 있습니다.

그래서 입력받은 문자를 Set 자료구조에 담아서 중복을 제거한 뒤에 ArrayList로 옮겨서 정렬해주었습니다.

	import java.util.*;
	import java.io.*;
	// https://www.acmicpc.net/problem/1181
	class Main {
	public static void main(String args[]) throws Exception {
	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	int N = Integer.parseInt(br.readLine());
	// 중복 제거를 위해 set으로 먼저 input
	HashSet<String> set = new HashSet<String>();
	for(int i=0; i<N; i++)
	set.add(br.readLine());
	// List 변환
	ArrayList<String> list = new ArrayList<String>(set);
	// Comparator 클래스를 통하여 custom 정렬
	// 길이에 따라서 먼저 정렬하고 길이가 같으면 사전순으로 정렬
	Collections.sort(list, new Comparator<String>() {
	public int compare(String v1, String v2) {
	if(v1.length() > v2.length())
	return 1;
	else if(v1.length() < v2.length())
	return -1;
	else
	return v1.compareTo(v2);
	}
	});
	for(String s : list)
	System.out.println(s);
	}
	}

view raw BOJ1181.java hosted with ❤ by GitHub