배열 만들기 4

문제 설명

정수 배열 arr가 주어집니다. arr를 이용해 새로운 배열 stk를 만드려고 합니다.

변수 i를 만들어 초기값을 0으로 설정한 후 i가 arr의 길이보다 작으면 다음 작업을 반복합니다.

만약 stk가 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.
stk에 원소가 있고, stk의 마지막 원소가 arr[i]보다 작으면 arr[i]를 stk의 뒤에 추가하고 i에 1을 더합니다.
stk에 원소가 있는데 stk의 마지막 원소가 arr[i]보다 크거나 같으면 stk의 마지막 원소를 stk에서 제거합니다.
위 작업을 마친 후 만들어진 stk를 return 하는 solution 함수를 완성해 주세요.

제한 사항

1 ≤ arr의 길이 ≤ 100,000
1 ≤ arr의 원소 ≤ 100,000

나의 코드

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        for(int i=0; i<arr.length; i++) {
            if(list.size() == 0) {
                list.add(arr[i]);
                i+=1;
            } else if(list.size() != 0 && list.size()-1 < arr[i]) {
                list.add(arr[i]);
                i+=1;
            } else if(list.size() != 0 && list.size()-1 >= arr[i]) {
                list.remove(list.size()-1);
            }
        }
        int[] stk = list.stream().mapToInt(i->i).toArray();
        return stk;
    }
}

첫 번째 코드 : 코드 실행 시 테스트 코드는 통과, 그러나 제출 후 채점하기에서 런타임 에러 발생 😭

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        for(int i=0; i<arr.length; i++) {
            if(list.size() == 0) {
                list.add(arr[i]);
                i+=1;
            } else {
                int last = list.get(list.size()-1);
                if(list.size() != 0 && last < arr[i]) {
                    list.add(arr[i]);
                    i+=1;
                } else if(list.size() != 0 && last >= arr[i]) {
                    list.remove(list.size()-1);
                }
            }
        }
        int[] stk = list.stream().mapToInt(i->i).toArray();
        return stk;
    }
}

두 번째 코드 : 마지막 인덱스 값과 arr[i]를 비교할 때 해당 리스트가 갖는 값을 비교해야할 것 같아서 int last = list.get(list.size()-1);를 추가했으나 이번엔 코드는 통과해도 채점하기에서 런타임 에러가 아닌 실패가 뜬다.

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        int i=0;
        while(i<arr.length) {
            if(list.size()==0) {
                list.add(arr[i]);
                i+=1;
            } else {
                int last = list.get(list.size()-1);
                if(list.size()!=0 && last<arr[i]) {
                    list.add(arr[i]);
                    i+=1;
                } else if(list.size()!=0 && last>=arr[i]) {
                    list.remove(list.size()-1);
                }
            }   
        }
        int[] stk = list.stream().mapToInt(k->k).toArray();
        return stk;
    }
}

최종 코드 : 다시 문제 내용을 차근차근 읽어보며 while문으로 코드를 짰더니 통과

다른 사람 코드

import java.util.*;

class Solution {
    public Stack<Integer> solution(int[] arr) {
        Stack<Integer> stack = new Stack<>();
        int i = 0;

        while (i < arr.length) {
            if (stack.empty() || stack.peek() < arr[i]) {
                stack.push(arr[i]);
                i++;
            } else if (stack.peek() >= arr[i]) {
                stack.pop();
            }
        }
        return stack;
    }
}

stack을 이용한 코드

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        ArrayList<Integer> arrayList = new ArrayList<>();
        int i = 0;
        while (i < arr.length) {
            if (arrayList.isEmpty()) {
                arrayList.add(arr[i]);
                i += 1;
            } else if (arrayList.get(arrayList.size() - 1) < arr[i]) {
                arrayList.add(arr[i]);
                i += 1;
            } else if (arrayList.get(arrayList.size() - 1) >= arr[i]) {
                arrayList.remove(arrayList.size() - 1);
            }
        }
        int idx = 0;
        int[] stk = new int[arrayList.size()];
        for (int data : arrayList)
            stk[idx++] = data;
        return stk;
    }
}

느낀 점

처음 만나보는 런타임 에러가 나서 당황했다.. 분명 맞는 것 같은 코든데 통과가 안되니 원... 😂😂 혹시라도 코드가 잘못된 걸 수도 있으니 런타임 에러 원인을 다시 한 번 고민해봐야겠다 ...