백준 17608번 - 막대기

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

17608번: 막대기

 

 

 

 

 

 

예전에 풀었던 백준의 탑 문제(https://www.acmicpc.net/problem/2493)와 비슷하다고 생각했다. 각각의 높이가 다르고, 어느 한쪽에서 보았을 때 뒤에 있는 것들이 보이는지를 판단해야 한다. 탑 문제의 해설을 들었을 때, 스택을 사용해서 풀 수 있다고 해서 이 문제도 스택으로 접근해서 풀었다.

 

 

 

 

 

오른쪽에서 보았을 때, 제일 오른쪽에 있는 것보다 큰 것만 남아있어야 한다.

스택에 h를 쌓는데, 후에 큰 h가 들어올 때마다 스택에서 h보다 작은 것들을 빼준다.

그리고 다 빼줬으면 h를 넣는다.

 

마지막에 스택의 사이즈를 출력하면 된다.

 

 

 

 

 

코드

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

public class Main {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine());
		
		Stack<Integer> stack = new Stack<>();

		int h = Integer.parseInt(br.readLine());
		stack.push(h);
		
		for(int i = 1; i < N; i++) {
			h = Integer.parseInt(br.readLine());
			
			while(true) {
				if(!stack.isEmpty() && h >= stack.peek()) {
					stack.pop();
				}else {
					stack.push(h);
					break;
				}
			}
			
		}
		
		System.out.println(stack.size());
	}

}

 

 

 

 

 

 

결과