프로그래밍 문제/BOJ(백준 온라인 저지)
[BOJ 1181번] 단어 정렬(JAVA)
테크케찰
2020. 9. 30. 01:47
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이라는 변수를 선언해 정렬되어있는 배열에서 중복되는 값은 하나만 출력하도록 설정했습니다.