[백준] 17608 막대기 - Data Structure / Java

• 문제 링크

17608번: 막대기

 

• 풀이 과정

stack에 각 막대기의 높이를 push하고 모두 pop한다.

peek하여 최대 높이가 이후 꺼낼 높이보다 높다면 cnt++를 수행하고,

높이는 최대값으로 갱신해 오른쪽에서 현재 최대 높이보다 낮은 막대기는 볼 수 없는 조건을 만족시킨다.

 

• 풀이 코드

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

public class Main {

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

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

        Stack<Integer> stk = new Stack<>();
        while (n-- > 0)
            stk.push(Integer.parseInt(br.readLine()));

        int max = stk.pop(), cnt = 1;
        while (!stk.empty()) {
            if (max < stk.peek())
                cnt++;

            max = Math.max(max, stk.pop());
        }

        bw.write(String.valueOf(cnt));
        bw.flush();
    }

}