[Java] PS에서 자주 사용하는 입력 클래스에 대해

알고리즘 문제풀이에서 표준 입력을 사용하는 방법은 시간이 오래 걸리기 때문에 좋은 전략은 아닙니다.

PS에서 자주 사용되는 자바 입력 클래스 사용법에 대해 간단히 알아봅시다.

 

대표적으로 자주 사용하는 입력 클래스에는 Scanner, BufferedReader가 있고, 입력받은 문자열을 토크나이징하여 슬라이스할 수 있는 StringTokenizer가, 여러 문자열들을 모아 출력 결과를 한 곳에 빌드할 수 있는 StringBuilder가 있습니다.

Scanner

next + 자료형을 이용해 입력을 받을 수 있고, hasNext + 자료형을 ㅁ이용해 입력받을 수 있는 자료형이 있는지 구할 수 있습니다.

 

기본적인 선언 및 사용

public class ScannerTest {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int a = sc.nextInt();
        double d = sc.nextDouble();
        String str = sc.next();

        System.out.println("a = " + a);
        System.out.println("d = " + d);
        System.out.println("str = " + str);
    }
}

hasNextXXX을 이용해 입력에서 정수가 주어지는 동안 계속 입력 받기 

public class ScannerTest1 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while (sc.hasNextInt()) {
            int k = sc.nextInt();
            System.out.println("k = " + k);
        }
    }
}

정수와 문자열 동시 처리

public class ScannerTest2 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int i = sc.nextInt(); // 정수 + '\n'
        // sc.nextLine();
        String s = sc.nextLine();  // s에 줄바꿈문자가 자동으로 들어가게 됩니다.
        System.out.println("i = " + i);
        System.out.println("s = " + s);
    }
}

이 코드에서 1을 입력하고 입력을 마치기 위해 엔터를 입력하면 버퍼에 '1\n'이 들어가게 됩니다.

이 때 String s에 버퍼의 줄바꿈문자가 삽입되고 종료됩니다. 따라서 아래와 같은 결과를 얻게 됩니다.

이를 방지하기 위해 줄바꿈을 입력받는 코드인 scanner.nextLine(); 를 작성해 주어야 올바른 의도대로 동작할 수 있습니다.

• Scanner의 단점 : 속도가 매우 느리고 입력을 많이 받아야 하는 경우 별도의 처리가 필요합니다.

 

BufferedReader

• Scanner의 느린 속도 개선된 입력 클래스
• read(), readLine()만 있기 떄문에 문자열이 아닌 정수형은 파싱이 필요하다는 단점이 있습니다.

public class BufferedReaderTest0 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] line = br.readLine().split(" ");

        int result = Integer.valueOf(line[0]) + Integer.valueOf(line[1]);

        System.out.println("result = " + result);
    }
}

 

StringTokenizer

• 입력받은 문자열을 토큰으로 잘라야 할 때 유용한 클래스
• BufferedReader에서 readLine().split(" ")으로도 가능

public class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
    String line = bf.readLine();
    StringTokenizer st = new StringTokenizer(line, " ");
    int sum = 0;
    while(st.hasMoreTokens())
      sum += Integer.valueOf(st.nextToken());
    System.out.println(sum);
  }
}

 

StringBuilder

• 출력해야 하는 것이 많은 경우 매번 출력하는 것보다, 한 번에 모아서 출력하는 것이 속도적으로 유리한데, StringBuilder를 사용하면 문자열을 만들 수 있습니다.

public class StringBuilderTest {
    public static void main(String[] args) {
        StringBuilder sb = new StringBuilder();

        for (int i = 1; i <= 5; i++) {
            sb.append(i + "\n");
        }

        System.out.println(sb);
    }
}

 

<참고 자료>

[Algorithm] 알고리즘을 위한 자바 Java IO