본문 바로가기
코딩테스트/백준

[백준 2751] 수 정렬하기 2 - Java

by naahy 2022. 12. 4.

문제

N개의 수가 주어졌을 때, 이를 오름차순으로 정렬하는 프로그램을 작성하시오.

 

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 수가 주어진다. 이 수는 절댓값이 1,000,000보다 작거나 같은 정수이다. 수는 중복되지 않는다.

 

출력

첫째 줄부터 N개의 줄에 오름차순으로 정렬한 결과를 한 줄에 하나씩 출력한다.

 



단순히 [ic]Arrays.sort()[/ic] 를 이용하면 시간초과가 나나? 싶어서 잠깐 고민했지만 의외로 쉽게 통과 돼서 놀랐다.
하지만 아무래도 더 간편한 방법이 있을 것 같아 찾아봤다.


int 배열 대신 boolean 배열을 사용한다.

 

풀이 1

  • 첫째 줄에서 입력 받은 N 만큼의 크기를 가지는 int 배열 생성
  • for문을 돌면서 int 배열에 입력받은 수를 저장한 후 [ic]Arrays.sort()[/ic]를 이용해 배열 오름차순 정렬

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuffer sb = new StringBuffer();
		
		int n = Integer.parseInt(br.readLine());
		
		int[] num = new int[n];
		
		for (int i=0; i<n; i++) {
			num[i] = Integer.parseInt(br.readLine());
		}
		
		Arrays.sort(num);
		
		for (int i:num) {
			sb.append(i).append('\n');
		}
		
		System.out.println(sb);
	}

}

 


 

풀이 2

  • int 배열 대신 2,000,001 크기의 boolean 배열 생성
    • 입력 되는 수는 절댓값이 1,000,000보다 작은 정수이므로
  • for문을 돌며 각 수를 입력 받고, [ic]배열[해당 수 + 1,000,000][/ic]의 값을 [ic]true[/ic]로 변경
  • 다시 boolean 배열을 돌며 true값을 갖는 인덱스 출력

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuffer sb = new StringBuffer();
		
		int n = Integer.parseInt(br.readLine());
		
		boolean[] num = new boolean[2_000_001];
		
		for (int i=0; i<n; i++) {
			num[Integer.parseInt(br.readLine()) + 1_000_000] = true;
		}
		
		for (int i=0; i<num.length; i++) {
			if (num[i])
				sb.append(i - 1_000_000).append('\n');	// 입력받은 수를 그대로 출력하기 위함
		}
		
		System.out.println(sb);
	}

}

 

 

 

 

시간이 단축된 것을 알 수 있다.

댓글