본문 바로가기
백준

[백준] 단계별로 풀어보기(단계:12,정렬,JAVA)1181번, 단어 정렬

by 열정적인 이찬형 2022. 1. 15.

문제 링크

1181번: 단어 정렬
 
www.acmicpc.net

주의사항

  • JAVA를 사용하여 프로그램을 사용하였습니다.
  • 백준에서 코드를 작성하였을 때 아래 형태에서 Main에서 결과가 출력되어야 합니다.
public class Main{ 	
	public static void main(String[] args){
    }
}

문제 설명


접근 방법

문제에 내용을 보고 리스트에 Collections.sort()를 진행하면 정렬이 될 것임을 파악하였습니다.

Comparator를 만들어서 정렬하도록 진행하였습니다.

  • BufferedReader를 사용하여 입력 값을 받았습니다.
  • String타입을 저장하는 ArrayList를 만들었습니다.
  • for문에서 list.contains()를 사용하여 중복되는지 확인 후 ArrayList에 값들을 저장하였습니다.
  • Collections.sort()에 Comparator를 추가하여 정렬을 진행하였습니다.
  • String 타입의 길이가 같으면 사전순으로 정렬하도록 하였습니다.
  • 그 이외에는 String 값의 길이를 비교하여 작은 순으로 정렬하도록 하였습니다.
  • for문을 통해 리스트에 값을 bw에 저장하였습니다.
  • BufferedWriter를 통해 저장된 결과를 출력하였습니다.

결과 코드

import java.io.*;
import java.util.*;
public class Main{
	public static void main(String[] args) throws IOException{
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        //BufferedReader를 통해 입력 값 받기
    	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
       	//BufferedWriter를 통해 결과 출력
    	int index = Integer.parseInt(br.readLine());	//입력 반복 횟수
    	ArrayList<String> list = new ArrayList<String>();	//리스트 초기화
    	for(int i=0;i<index;i++) {
    		String text = br.readLine();
    		if(!list.contains(text))	//입력 값 중복되는지 확인후 리스트에 값 저장
    			list.add(text);
    	}
    	Collections.sort(list, new Comparator<String>() {	//정렬
    		@Override
    		public int compare(String s1, String s2) {	//Comparator 구성
    			if(s1.length()==s2.length())	//글자수가 같을 때
    				return s1.compareTo(s2);
    			else		//그 외에
    				return s1.length()-s2.length();
    		}
		});
    	for(int i=0;i<list.size();i++) {	//리스트 값 BufferedWriter에 저장
    		bw.write(list.get(i) + "\n");
    	}
    	bw.flush();					//결과 출력
    	bw.close();
    	br.close();
	}
}

댓글