14. 17298번 오큰수

 

 

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 training2 {

	public static void main(String[] args) throws Exception {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        //스택에는 인덱스 값이 저장될 것이다
        Stack<Integer> stack = new Stack<>();

        //입력받는 명령의 수
        int N = Integer.parseInt(br.readLine());
		
        // 명령받는 3,5,2,7 과 같은 수들이 들어갈 배열
        int[] A = new int[N];

        // 결과값 5,7,7,-1이 들어갈 배열
        int[] B = new int[N];


        for (int i = 0; i < N; i++){
            B[i] = -1;
        }
        //결과 배열의 모든값을 -1로 초기화 

        // 3 5 2 7 을 수로 받아 A배열에 넣는 과정
        String[] arr = br.readLine().split(" ");
        A[0] = Integer.parseInt(arr[0]);
        stack.push(0);
        

        for (int i = 1; i < N; i++) {
            //입력받은 수를 A배열에 넣는다. 
            int n = Integer.parseInt(arr[i]);
            //스트링으로 받아온 배열이기에 형변환
            A[i] = n;

            
            while (!stack.isEmpty() && A[stack.peek()] < n) {
                //스택의 가장 윗부분의 값이 입력받은 수보다 작을때까지 반복
                // i번째와 i-1 번째의 값을 비교하는 것
                B[stack.pop()] = n;  
            }
            stack.push(i);
        }

        for (int i = 0; i < N; i++){
            bw.write(B[i] + " ");
            //결과값
        }
         
        bw.flush();
        bw.close();
    }
}

풀고나서 정리하는 지금 다시봐도 굉장히 혼란스럽다. 

스택에 인덱스값을 넣어 현재 탐색하고있는 수열의 값과 비교한 후 

스택에 들어있는 인덱스값을 이용해서 수열을 호출해서 결과를 만들어 낸다.

추가적인 공부 이후 완벽한 이해가 필요하다.

 

 

결과