[Baekjoon] 30803_수도꼭지

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

---

< 수도꼭지 >

 

문제 풀이 

 

수도꼭지가 잠겼을 때와 열렸을 때를 구분하여 탱크에 담기는 물의 양을 구한다.

 

 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 _30803_ { // 수도꼭지

	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 N = Integer.parseInt(bf.readLine());
		int arr[] = new int[N + 1];
		boolean visited[] = new boolean[N + 1];

		long answer = 0;
		st = new StringTokenizer(bf.readLine());
		for (int i = 1; i < N + 1; i++) {
			arr[i] = Integer.parseInt(st.nextToken());
			answer += arr[i];
		}
		bw.write(answer + "\n");

		int Q = Integer.parseInt(bf.readLine());
		for (int i = 0; i < Q; i++) {
			st = new StringTokenizer(bf.readLine());

			int num = Integer.parseInt(st.nextToken());
			if (num == 1) {
				int temp = Integer.parseInt(st.nextToken());
				int x = Integer.parseInt(st.nextToken());
				if (visited[temp]) {
					arr[temp] = x;
				} else {
					answer -= arr[temp];
					arr[temp] = x;
					answer += arr[temp];
				}
			} else {
				int x = Integer.parseInt(st.nextToken());
				if (!visited[x]) {
					visited[x] = true;
					answer -= arr[x];
				} else {
					visited[x] = false;
					answer += arr[x];
				}
			}
			bw.write(answer + "\n");
		}
		bw.flush();
	}
}

변수)
N : 수도꼭지의 수
arr : 수도꼭지에서 나오는 물의 양
visited : 수도꼭지 열렸는지 여부
answer : 1분 동안 탱크에 담기는 물의 양
Q : 조작의 수
num : 조작 번호

 

수도꼭지의 수를 입력받는다. 수도꼭지의 수만큼 수도꼭지에서 나오는 물의 양을 입력받아 arr에 저장한다. 조작의 수 Q를 입력받는다. 조작의 수만큼 조작을 입력받아 다음 과정을 반복한다.

 

1) 조작 번호가 1이라면 수도꼭지 번호 temp와 물의 양 x를 입력받는다. 만약 temp번 수도꼭지가 잠겨있다면 물의 양만 바꾸고, 열려있다면 answer에 원래 물의 양을 빼주고 x를 더한다.

2) 조작 번호가 2라면 수도꼭지 번호 x를 입력받는다. 만약 x번 수도꼭지가 열려있다면 잠그고 answer에 그 값을 빼준다. 만약 잠겨있다면 열고 answer에 그 값을 더한다.

3) answer을 출력한다.

---

---