인프런 커뮤니티 질문&답변

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

자바로 작성하니까 시간초과가 나는데, 상관없는 건가요??

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] argvs) {
		Scanner sc = new Scanner(System.in);
		
		int n=sc.nextInt();
		int[] num = new int[n+1];
		
		for(int i=1; i<=n; i++) {
			num[i] = sc.nextInt();
		}
		
		Stack<Integer> st = new Stack<>();
		int[] answer = new int[n+1];
		
		for(int i=n; i>=1; i--) { //제일 뒤에서 부터 스택에 넣는다.(스택에 넣는 것은 [[인덱스 번호]]를 넣는다)
			while(!st.isEmpty() && num[i]>num[st.peek()]) { //num배열에서 인덱스 번호에 해당하는 것을 본다. 현재 배열의 값이 스택의 top부분보다 크면 pop한다. 다시말해 스택 안은 오름차순 정렬된다.
				answer[st.peek()] = i; //정답 배열에 인덱스 값 넣기, st.peek도 인덱스, i도 인덱스 번호
				st.pop();
			}
			st.push(i); //스택에 삽입
		}
		
		for(int i=1; i<=n; i++) { //답 출력
			System.out.print(answer[i]+" ");
		}
	}
}