백준 6198번 옥상 정원 꾸미기 (JAVA)

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

6198번: 옥상 정원 꾸미기

스택을 사용하여 풀 수 있는 문제였다. 비슷한 문제의 난이도 상위 버전으로 히스토그램이 있으니, 풀어보는 것을 추천한다.

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

public class Main {
	static Function<String, Integer> stoi = Integer::parseInt;
	static int N;

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		N = stoi.apply(br.readLine());
		long result = 0;
		int tmp;
		Stack<Integer> stack = new Stack<>();
		for (int i = 0; i < N; i++) {
			tmp = stoi.apply(br.readLine());
			while (!stack.isEmpty() && stack.peek() <= tmp) {
				stack.pop();
			}
			result += stack.size();
			stack.add(tmp);
		}
		System.out.println(result);
	}

}