성능 요약
메모리: 0.0 MB, 시간: 0.00 ms
구분
코딩테스트 연습 > 연습문제
채점결과
정확성: 100.0
합계: 100.0 / 100.0
제출 일자
2024년 07월 02일 22:52:03
문제 설명
정수로 이루어진 배열 numbers가 있습니다. 배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 뒷 큰수라고 합니다.
정수 배열 numbers가 매개변수로 주어질 때, 모든 원소에 대한 뒷 큰수들을 차례로 담은 배열을 return 하도록 solution 함수를 완성해주세요. 단, 뒷 큰수가 존재하지 않는 원소는 -1을 담습니다.
제한사항
- 4 ≤
numbers의 길이 ≤ 1,000,000- 1 ≤
numbers[i]≤ 1,000,000
- 1 ≤
입출력 예
| numbers | result |
|---|---|
| [2, 3, 3, 5] | [3, 5, 5, -1] |
| [9, 1, 5, 3, 6, 2] | [-1, 5, 6, 6, -1, -1] |
입출력 예 설명
입출력 예 #1
2의 뒷 큰수는 3입니다. 첫 번째 3의 뒷 큰수는 5입니다. 두 번째 3 또한 마찬가지입니다. 5는 뒷 큰수가 없으므로 -1입니다. 위 수들을 차례대로 배열에 담으면 [3, 5, 5, -1]이 됩니다.
입출력 예 #2
9는 뒷 큰수가 없으므로 -1입니다. 1의 뒷 큰수는 5이며, 5와 3의 뒷 큰수는 6입니다. 6과 2는 뒷 큰수가 없으므로 -1입니다. 위 수들을 차례대로 배열에 담으면 [-1, 5, 6, 6, -1, -1]이 됩니다.
출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges
풀이
Java
import java.util.Stack;
import java.util.Arrays;
class Solution {
public int[] solution(int[] numbers) {
Stack<Integer> stack = new Stack<Integer>();
//-1로 원소 채우기
int[] result = new int[numbers.length];
Arrays.fill(result, -1);
//뒤에서부터 검사
for (int i=numbers.length-1; i>-1; i--) {
//스택이 비어있지 않고 현재 숫자가 맨 위 원소 이상이면
while (!stack.empty() && numbers[i] >= stack.peek()) {
stack.pop();
}
//스택이 비어있지 않은 경우
if (!stack.empty()) {
result[i] = stack.peek();
}
//스택에 현재 숫자 추가
stack.push(numbers[i]);
}
return result;
}
}