[programmers] 배열 만들기 6 - Java

문제 설명

• 만약 stk이 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.

• stk에 원소가 있고, stk의 마지막 원소가 arr[i]와 같으면 stk의 마지막 원소를 stk에서 제거하고 i에 1을 더합니다.

• stk에 원소가 있는데 stk의 마지막 원소가 arr[i]와 다르면 stk의 맨 마지막에 arr[i]를 추가하고 i에 1을 더합니다.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000,000
• arr의 원소는 0 또는 1 입니다.

입출력 예

입출력 예 설명

• 각 작업을 마친 후에 배열의 변화를 나타내면 다음 표와 같습니다.

• 따라서 [0, 1, 0]을 return 합니다.

• 각 작업을 마친 후에 배열의 변화를 나타내면 다음 표와 같습니다.

• 따라서 [0, 1, 0, 1, 0]을 return 합니다.

• 각 작업을 마친 후에 배열의 변화를 나타내면 다음 표와 같습니다.

• 마지막에 빈 배열이 되었으므로 [-1]을 return 합니다.

solution.java

import java.util.*;

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

핵심 키워드

• 정수형 ArrayList를 배열로 변환하려면 다음과 같은 형식으로 코드를 작성한다.
• Arrays.stream(대상 배열)을 통해 대상 배열의 값에 하나하나 접근할 수 있다.
• .mapToInt() 해당 요소에 주어진 함수를 적용한 결과로 구성된 IntStream을 반환한다.
• .toArray() 메소드를 통해 해당 요소가 포함된 배열을 반환한다.

• ArrayList에서 값을 가져오려면 list.get(인덱스 위치) 메소드를 사용한다.

결론!