뒤에 있는 큰 수 찾기
정수로 이루어진 배열 numbers
가 있습니다. 배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 뒷 큰수라고 합니다.
정수 배열 numbers
가 매개변수로 주어질 때, 모든 원소에 대한 뒷 큰수들을 차례로 담은 배열을 return 하도록 solution 함수를 완성해주세요. 단, 뒷 큰수가 존재하지 않는 원소는 -1을 담습니다.
https://school.programmers.co.kr/learn/courses/30/lessons/154539
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
public class Main {
public static void main(String[] args) {
int[] number = {9, 1, 5, 3, 6, 2};
var reuslt = solution(number);
System.out.println(Arrays.toString(reuslt));
}
public static int[] solution(int[] numbers) {
int[] result = new int[numbers.length];
int max = numbers[numbers.length-1];
for (int i = numbers.length-1; i > -1; i--) {
int current = numbers[i];
result[i] = -1;
if (current > max) {
max = current;
continue;
}
for (int l = i+1; l < numbers.length; l++) {
if (current >= max) {
break;
}
if (current < numbers[l]) {
result[i] = numbers[l];
break;
}
}
}
return result;
}
}
시간초과
import java.util.*;
import java.util.stream.Collectors;
public class Main {
public static void main(String[] args) {
int[] number = {9, 1, 5, 3, 6, 2};
var reuslt = solution(number);
System.out.println(Arrays.toString(reuslt));
}
public static int[] solution(int[] numbers) {
int[] result = new int[numbers.length];
Arrays.fill(result, -1);
List<Integer> notFound = new ArrayList<>();
for (int i = 0; i < numbers.length; i++) {
int mk =notFound.size()-1;
while (!notFound.isEmpty() && numbers[notFound.get(notFound.size()-1)] < numbers[i]) {
if (mk == -1) {
break;
}
if (numbers[notFound.get(mk)] < numbers[i]) {
result[notFound.get(mk)] = numbers[i];
notFound.remove(mk);
}
mk --;
}
notFound.add(i);
}
return result;
}
}