[백준] 8958번 : OX퀴즈 - JAVA [자바]

 

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

8958번: OX퀴즈

 

---

• 문제

 

매우 간단한 문제다!

 

※ 주의할 점

1. 각 케이스마다 점수를 출력한다.
2. O 의 연속 개수 만큼 점수를 누적하여 합산하고 X가 나오면 누적값은 0으로 초기화 된다.

 

 

---

• 3가지 풀이 방법을 제시한다.

먼저 가장 기본적인 배열을 이용한 방법을 사용할 것이다.

문자열 배열인 String 을 통해 각 줄의 OX 문자열을 각 배열에 저장한 뒤 이후 반복, 조건문을 통해 풀 것이다.

 

이렇게 배열을 이용하여 입력 방법을 달리하여 두 가지 방법으로 풀어볼 것이다.

 

 

그리고 다른 방법으로는 배열을 이용하지 않는 방법을 제시 할 것이다.

---

• 풀이

- 방법 1 

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);

		String arr[] = new String[in.nextInt()];

		for (int i = 0; i < arr.length; i++) {
			arr[i] = in.next();
		}
		
		in.close();
		
		for (int i = 0; i < arr.length; i++) {
			
			int cnt = 0;	// 연속횟수
			int sum = 0;	// 누적 합산 
			
			for (int j = 0; j < arr[i].length(); j++) {
				
				if (arr[i].charAt(j) == 'O') {
					cnt++;
				} 
				else {
					cnt = 0;
				}
				sum += cnt;
			}
			
			System.out.println(sum);
		}
	}
}

 

가장 기본적인 방법이라 할 수 있겠다.

 

먼저 String 배열 arr 의 각각의 원소에 문자열을 저장해준다.

 

그리고 cnt 는 몇 번 연속했는지,

sum 은 누적 합산 값을 나타내는 변수로 선언을 해준다.

 

그리고 그 안에 for문으로 한 원소당 String 의 길이만큼 반복하여 해당 원소의 문자열을 charAt() 메소드를 통해 하나하나 검사한다.

그 과정에서 만약 O 라는 문자가 나타나면 cnt 을 1 증가시키고 아닐경우는 X 문자이니 cnt 를 0 으로 초기화 시켜준다.

 

그렇게 하여 sum에 cnt 값을 누적하여 더해준 뒤 해당 배열 원소의 문자열을 다 검사하고나면 sum 을 출력해주면 된다.

---

 

- 방법 2

 

위에서는 Scanner 을 사용하여 풀었다. 위 방법 말고 BufferedReader 을 통해 풀어보고

좀 더 빠른 실행을 위해 System.out.println() 을 반복하여 호출하지 않고 StringBuilder 를 이용하여 문자열을 이어준 뒤 한 번에 출력을 해보고자 한다.

 

기본 알고리즘은 같다.

 

 

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		StringBuilder sb = new StringBuilder();
		int test_case = Integer.parseInt(br.readLine());	//테스트 케이스
		
		String arr[] = new String[test_case];

		for (int i = 0; i < test_case; i++) {
			arr[i] = br.readLine();
		}

		
		for (int i = 0; i < test_case; i++) {
			
			int cnt = 0;	// 연속횟수
			int sum = 0;	// 누적 합산 
			
			for (int j = 0; j < arr[i].length(); j++) {
				
				if (arr[i].charAt(j) == 'O') {
					cnt++;
				} else {
					cnt = 0;
				}
				sum += cnt;
			}
			
			sb.append(sum).append('\n');
		}
		
		System.out.print(sb);
	}
}

 

단, 맨 처음 테스트케이스 수를 입력받을 때는 BufferedReader 가 문자열을 입력받기 때문에 Integer.parseInt() 로 String 에서 int 로 변환 시켜주어야 한다.

 

그리고 결과를 저장할 때 StringBuilder 에 문자열을 이어주고 반드시 개행('\n') 을 해주도록 한 뒤, 프로그램 마지막에 sb로 객체를 출력해주도록 한다.

 

 

 

---

- 방법 3 

 

배열을 사용하지 않고 하는 방법이다.

위에서는 String이라는 배열을 사용했지만, 사실 getBytes() 메소드와 for-each 문을 쓰면 매우 쉽고 간결한 코드가 될 수 있다.

 

 

getBytes() 메소드는 입력 문자열을 byte 단위의 배열로 반환시켜주는 메소드다.

그리고 '배열'로 반환을 하므로 for-each 문에서도 쓸 수 있다.

 

일단 아래 코드를 보자.

 

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		StringBuilder sb = new StringBuilder();

		int test_case = Integer.parseInt(br.readLine()); // 테스트 케이스

		for (int i = 0; i < test_case; i++) {

			int cnt = 0; // 연속횟수
			int sum = 0; // 누적 합산

			for (byte value : br.readLine().getBytes()) {
				
				if (value == 'O') {
					cnt++;
					sum += cnt;
				} 
				else {
					cnt = 0;
				}
			}
			
			sb.append(sum).append('\n');

		}
		System.out.println(sb);

	}
}

for( byte value : br.readLine().getBytes() ) 라는 for-each 구문을 보자.

 

먼저 .br.readLine() 은 한 줄을 문자열로 입력을 받는다. 그리고 getBytes() 는 입력받은 문자열을 byte 형 배열로 반환한다.

즉, 위의 for-each 문은 입력을 한 줄 받으면 해당 문자열의 길이만큼 반복하면서 해당 문자열의 문자를 하나씩 value 에 저장하여 꺼내오는 것이다.

 

위와 같은 방법으로 직접 배열을 선언하여 쓰지 않고도 가능다는 것을 보여주고 싶었다.

---

• 성능
  
  

위에서 부터 순서대로

 

채점 번호 : 18285292  -  BufferedReader + 배열 X

채점 번호 : 18285286  -  BufferedReader + 배열

채점 번호 : 18285275  -  Scanner + 배열

 

 

입력의 경우는 확실히 Scanner 보다는 BufferedReader 가 빠른 걸 볼 수 있다.

 

 

---

• 정리

BufferedReader 을 필자는 자주 애용하기에 다른 포스팅에서도 BufferedReader 을 쓴 코드들을 많이 볼 수 있지만, getBytes() 메소드를 직접 언급한 것은 아마 처음일 것이다.

 

해당 메소드는 String문자열을 byte[] 배열로 반환하기 때문에 '한글'을 쓰고자 한다면 toCharArray()를 쓰거나 charAt()을 써야하니 주의해서 쓰도록 하자.

 

이렇게 메소드 하나를 더 알아 가셨으면 좋겠다.