[Java] StringBuilder 로 출력하여 시간초과 해결

상황 설명

 

코딩테스트 준비로 알고리즘 문제를 풀다가 System.out.println을 사용하여 출력하였더니 시간초과가 발생하였다. 이런 경우에는 StringBuilder를 통해서 출력하면 실행시간을 현격히 줄일 수 있다는 것을 알게 되었다. 해당 문제는 아래에 첨부한다.

 

 

StringBuilder로 출력

 

// StringBuilder 선언
StringBuilder sb = new StringBuilder();

// StringBuilder에 출력하고 싶은 값 추가하기
for(Integer integer : arr){
    sb.append(integer);
}

// StringBuilder 출력
System.out.println(sb);

 

 

해당 문제와 코드 비교

 

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

2751번: 수 정렬하기 2

 

• System.out.println 으로 출력 => 시간초과

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class Main{
    public static void main(String[] args) {

        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        ArrayList<Integer> arr = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            arr.add(scanner.nextInt());
        }

        Collections.sort(arr);

        for (int i : arr) {
            System.out.println(i);
        }
    }
}

 

• StringBuilder로 모아서 출력 => 시간초과 해결!

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class Main{
    public static void main(String[] args) {

        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        ArrayList<Integer> arr = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            arr.add(scanner.nextInt());
        }
        Collections.sort(arr);
        StringBuilder sb = new StringBuilder();
        for (Integer integer : arr) {
            sb.append(integer);
            sb.append('\n');
        }
        System.out.print(sb);
    }
}