본문 바로가기

Java/프로그래머스

[JAVA] 프로그래머스 - 배열 만들기 4

https://school.programmers.co.kr/learn/courses/30/lessons/181918

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

 

 

정답 

 

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr) {
        ArrayList<Integer> list = new ArrayList<>();

        int i =0;
        while(i<arr.length)
        {  
            if(list.size()==0 || list.get(list.size()-1) <arr[i])
            {
                list.add(arr[i]);
                i++;
            }
            else
            {
                list.remove(list.size()-1);
            }
        }

        int stk[] = new int[list.size()];

        for(int j=0; j<list.size(); j++)
        {
            stk[j] = list.get(j);
        }
        return stk;
    }
}

 

 

다른 사람의 풀이 

 

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;
    }
}