티스토리 뷰
728x90
https://www.acmicpc.net/problem/17298
스택 Java
주어진 수열에서 각 원소마다 본인의 오른쪽에서 가장 가까운 본인보다 큰 수를 출력하는 문제
접근방법
이 문제의 포인트는 바로 스택에 오큰수가 될 수 있는 후보들을 저장하고 있다고 생각하고
스택에서 위에 있는 수가 아래에 있는 수보다 항상 커야한다는 점을 생각해야한다!
따라서 스택의 top이 본인보다 작다면 pop해줘야 한다. 스택에는 나(top이 될)보다 큰 값들만 들어있어야 한다.
그렇지 않으면 내 오른쪽에 있는 모든 수를 탐색하는 것과 다를게 없다.
오답노트
접근방법에 적혀있는 내용을 깨닫지 못하고 스택 두개를 사용해서 풀었는데 당연히 시간초과가 났다.
그리고 두 스택을 이용해서 풀때 stack = stack 하다가 stack이 참조 변수라는걸 망각해서,,clear하고 addAll해줬었다.
필요없게 됐지만.
arraylist set이 안돼서 거꾸로 넣고 reverse해줬다.
소스코드
import java.io.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Stack;
public class S17298_1 {
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());
String[] stringArr = br.readLine().split(" ");
int[] arr = new int[n];
for(int i = 0 ; i < n ; i++){
arr[i] = Integer.parseInt(stringArr[i]);
}
ArrayList<Integer> result = new ArrayList<>();
Stack<Integer> stack = new Stack<>();
for(int i = n-1 ; i >= 0 ; i--){
result.add(n-1-i,-1);
while(!stack.empty()){
if(arr[i] < stack.peek()){
result.set(n-1-i,stack.peek());
break;
}else{
stack.pop(); //나보다 크지 않다면 후보에서도 탈락(스택에서 빠짐)
}
}
stack.add(arr[i]);
}
Collections.reverse(result);
bw.write(result.toString().substring(1,result.toString().length()-1).replaceAll(",",""));
br.close();
bw.close();
}
}
728x90
'알고리즘 문제풀이' 카테고리의 다른 글
[백준 10866] 알고리즘 103일차 : 덱 (0) | 2021.07.09 |
---|---|
[백준 12015] 알고리즘 102일차 : 가장 긴 증가하는 부분 수열 2 (0) | 2021.07.08 |
[백준 5430] 알고리즘 10^2일차 : AC (0) | 2021.07.06 |
[백준 1966] 알고리즘 99일차 : 프린터 큐 (0) | 2021.07.05 |
[백준 11656] 알고리즘 98일차 : 접미사 배열 (0) | 2021.07.02 |
250x250
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 프로그래머스
- 백준
- 브루트포스
- 그리디알고리즘
- 이분탐색
- 자바
- 정렬
- 알고리즘
- dfs
- BFS
- 다이나믹프로그래밍
- 스택
- 투포인터
- 가장 큰 수 Swift
- Swift
- 최단경로
- c++
- 트리
- 토마토
- 가장 큰 수 프로그래머스
- 수학
- 웹크롤링
- dp
- 게임이론
- 우선순위큐
- 최대힙
- 백트래킹
- 파이썬
- 동적계획법
- 최소힙
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
글 보관함