[백준][Java] 2493번 탑 (스택, 구현)

[백준][Java] 2493번 탑 (스택, 구현)

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

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{ int N = stoi(br.readLine()); String [] inputStr = br.readLine(). split ( " " ); int [] height = new int [N + 1 ]; for ( int i = 1 ; i < = N; i + + ) { height[i] = stoi(inputStr[i - 1 ]); } Stack < int [] > st = new Stack < > (); // bw.write("0 "); // st.push(new int[] {height[1],1}); for ( int i = 1 ; i < = N; i + + ) { int curr = height[i]; while ( true ) { if (st.isEmpty()) { bw.write( "0 " ); st.push( new int [] {curr, i}); break ; } if (st.peek()[ 0 ] < curr) st.pop(); else { bw.write(st.peek()[ 1 ] + " " ); st.push( new int [] {curr, i}); break ; } } } 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/303 by ccl(A) rewrite - 2021-09-27 07:01:58