[Baekjoon] 9354_It Is Cold

문제(출처: https://www.acmicpc.net/problem/9354)

---

< It Is Cold >

 

문제 풀이 

 

뒤에서부터 확인하면서 A이면 (-), T이면 (+)로 계산한다. 이때, 계산값이 (-)가 된다면 0으로 초기화한다. 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class _9354_ { // It Is Cold

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		int t = Integer.parseInt(bf.readLine());

		for (int i = 0; i < t; i++) {
			int n = Integer.parseInt(bf.readLine());

			int arr[] = new int[n];

			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < n; j++) {
				arr[j] = Integer.parseInt(st.nextToken());
			}

			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < n; j++) {
				char dir = st.nextToken().charAt(0);

				if (dir == 'A') {
					arr[j] = -arr[j];
				}

			}

			long result = 0;
			if (arr[n - 1] > 0) {
				result = arr[n - 1];
			}

			for (int j = n - 2; j >= 0; j--) {
				result += arr[j];

				if (result < 0) {
					result = 0;
				}
			}
			bw.write(result + "\n");
		}
		bw.flush();
	}
}

변수)
t : 팀의 수
n : 선풍기 개수
arr : 선풍기 속도
dir : 선풍기 방향
result : 팀에 도달하는 바람의 속도

 

팀의 개수를 입력받는다. 팀의 개수만큼 다음 과정을 반복한다.

 

1) 선풍기 개수를 입력받는다.

2) 선풍기 개수만큼 선풍기 속도와 방향을 입력받는다. 방향이 A라면 속도를 음수로 저장한다.

3) result를 0으로 초기화하고 만약 마지막 선풍기의 속도가 양수라면 값을 업데이트한다. 

4) 뒤에서부터 선풍기를 탐색하며 result에 속도를 더한다. 이때, result 값이 음수가 된다면 0으로 업데트한다. 

5) 최종 result를 출력한다. 

---

---