본문 바로가기
프로그래밍 문제/BOJ(백준 온라인 저지)

[BOJ 1181번] 단어 정렬(JAVA)

by 테크케찰 2020. 9. 30.

www.acmicpc.net/problem/1181

 

1181번: 단어 정렬

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

www.acmicpc.net

import java.io.*;
import java.util.*;

class Main{
   
   static BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
   static BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
   static StringBuilder sb=new StringBuilder();
   static int N;
   static String[] arr;
   
   public static void main(String args[]) throws Exception {
	   N=Integer.parseInt(br.readLine());
	   arr=new String[N];
	   for(int i=0;i<N;i++) {
		   arr[i]=br.readLine();
	   }
	   Arrays.sort(arr, new Comparator<String>() {

		@Override
		public int compare(String o1, String o2) {
			int temp=o1.length()-o2.length();
			if(temp==0) {
				return (o1+o2).compareTo(o2+o1); //a.compareTo(b): a와 b가 같으면 0, a>b면 양수, a<b면 음수 
			}
			else return temp;
		}
		   
	   });
	   String temp_str="";
	   for(String s:arr) {
		   if(!temp_str.equals(s))	sb.append(s+"\n");
		   temp_str=s;
	   }
	   System.out.println(sb);
   }
}

Comparator를 이용해 문자열을 정렬했습니다.

문자열의 길이를 비교해 문자열 길이가 같으면 사전 순으로 정렬하고 문자열의 길이가 다르면 문자열의 길이 순서대로 정렬하도록 코딩을 짜보았습니다.

그리고 마지막에 temp_str이라는 변수를 선언해 정렬되어있는 배열에서 중복되는 값은 하나만 출력하도록 설정했습니다.