[백준][Java] 14719번 빗물 (시뮬레이션)

[백준][Java] 14719번 빗물 (시뮬레이션)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67

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 { private static BufferedReader br = new BufferedReader( new InputStreamReader( System . in )); private static BufferedWriter bw = new BufferedWriter( new OutputStreamWriter( System . out )); public static void main( String [] args) throws IOException{ br.readLine(); String [] str = br.readLine(). split ( " " ); int [] blockArr = new int [str. length ]; for ( int i = 0 ; i < str. length ; i + + ) blockArr[i] = stoi(str[i]); int res = 0 ; Stack < Integer > st = new Stack < > (); int leftBlock = 0 ; for ( int curr : blockArr) { if (curr > = leftBlock) { while ( ! st.isEmpty()) { res + = (leftBlock - st.pop()); } leftBlock = curr; } else { st.push(curr); } } int rightBlock = 0 ; Stack < Integer > finSt = new Stack < > (); while ( ! st.isEmpty()) { int curr = st.pop(); if (curr > = rightBlock) { while ( ! finSt.isEmpty()) { res + = (rightBlock - finSt.pop()); } rightBlock = curr; } else { finSt.push(curr); } } while ( ! finSt.isEmpty()) { res + = (rightBlock - finSt.pop()); } System . out . println (res); // bw.write(""); // bw.flush(); // bw.close(); } private static int stoi( String input) { return Integer. parseInt (input); } } Colored by Color Scripter

from http://aig2029.tistory.com/349 by ccl(A) rewrite - 2021-10-22 15:27:22