[Baekjoon] 17952_과제는 끝나지 않아!

Silver III

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

---

< 과제는 끝나지 않아! >

 

문제 풀이 

 

Stack을 사용해서 0이 입력될 때는 stack에서 값을 pop해와 시간을 -1 시켜 다시 stack에 넣어주며 만약 시간이 1이라면 과제를 끝낼 수 있으므로 점수를 총합에 더해준다. 만약 1이 입력될 때는 stack에 값을 넣어준다.

 

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class _17952_ { // 과제는 끝나지 않아!

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

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

		Stack<int[]> stack = new Stack<>();
		int answer = 0;

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());
			int num = Integer.parseInt(st.nextToken());
			if (num == 0) {
				if (!stack.isEmpty()) {
					int temp[] = stack.pop();
					if (temp[1] == 1)
						answer += temp[0];
					else
						stack.add(new int[] { temp[0], temp[1] - 1 });
				}
			} else {
				int a = Integer.parseInt(st.nextToken());
				int t = Integer.parseInt(st.nextToken());

				if (t == 1)
					answer += a;
				else
					stack.add(new int[] { a, t - 1 });
			}
		}
		System.out.println(answer);
	}
}

 

Main

변수)
n : 주어진 시간
stack : Stack<[과제 점수, 남은 시간]> 
answer : 과제 점수 총합
num: 0 or 1

 

- 주어진 시간(n) 입력

- 주어진 시간(n)만큼 과제 입력

: 0이면 과제가 주어지지 않으므로 과제 이어서 진행 -> 만약 1분 남았으면 과제가 끝나므로 총합(answer)에 점수 추가, 1분보다 많이 남았다면 남은 시간-1 해서 stack에 추가

: 1이면 과제가 주어지며 과제를 받자마자 시작하므로 시간-1 해서 stack에 추가 (주어지는 과제가 1분이라면 바로 종료되므로 총합(answer)에 추가)

---

---