[백준] 10989 수 정렬하기 3 Java

문제

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

 

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 10,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 수가 주어진다. 이 수는 10,000보다 작거나 같은 자연수이다.

 

출력

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

 

코드

import java.awt.*;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        int N = Integer.parseInt(br.readLine());
        int[] arr = new int[N];
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(br.readLine());
        }

        Arrays.sort(arr);
        for (int i = 0; i < N; i++) {
            sb.append(arr[i]).append("\n");
        }

        System.out.println(sb);
    }
}

문제보고 그냥 배열에 저장해서 정렬하고 출력하면 되지 않나 해서 그대로 구현했는데 시간초과 떴다.

그래서 StringBuilder 선언해서 sb 출력하니까 성공하긴 했는데 검색해보니 정렬하지 않고 성공하는 더 괜찮은 방법이 있었다.

 

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

public class Main {
    public static void main(String[] args) throws IOException {
        int[] arr = new int[10001];
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int N = Integer.parseInt(br.readLine());

        for (int i = 0; i < N; i++) {
            arr[Integer.parseInt(br.readLine())]++; //입력받은 숫자있는 배열의 카운트 +1
        }

        for (int i = 0; i < 10001; i++) {
            while (arr[i] > 0) { //입력받은 수가 있다는 뜻
                bw.write(i + "\n");
                arr[i]--;   //하나 출력해줬으니 없애줘야 함
            }
        }

        bw.flush();
    }
}

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

public class Main {
    public static void main(String[] args) throws IOException {
        int[] arr = new int[10001];
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        int N = Integer.parseInt(br.readLine());

        for (int i = 0; i < N; i++) {
            arr[Integer.parseInt(br.readLine())]++; //입력받은 숫자있는 배열의 카운트 +1
        }

        for (int i = 0; i < 10001; i++) {
            while (arr[i] > 0) { //입력받은 수가 있다는 뜻
                sb.append(i).append("\n");
                arr[i]--;   //하나 출력해줬으니 없애줘야 함
            }
        }

        System.out.println(sb);
    }
}

 

확실히 시간이 더 단축되긴 했다. 주의할 점은 그냥 print하면 또 시간초과나니까 BufferedWriter나 StringBuilder를 사용해줘야 한다.

 

77498786 - 정렬 o

77498704 - 정렬 x, StringBuilder

77498606 - 정렬 x, BufferedWriter

 

문제 출처

https://www.acmicpc.net/problem/10989