프로그래머스 Lv.0 배열 만들기 6

프로그래머스 Lv.0 배열 만들기 6

---

문제 설명

0과 1로만 이루어진 정수 배열 arr가 주어진다.

arr를 이용해 새로운 배열 stk를 만드려고 한다.

i의 초기값을 0으로 설정하고 i가 arr의 길이보다 작으면 다음을 반복한다.

- 만약 빈 배열이라면 arr[i]를 추가하고 i에 1을 더한다.

- 원소가 있고, 마지막 원소가 arr[i]와 같으면 마지막 원소를 제거하고 i에 1을 더한다.

- 원소가 있는데 마지막 원소가 arr[i]와 다르면 맨 마지막에 arr[i]을 추가하고 i에 1을 더한다.

위 작업을 마친 후 만들어진 stk를 return하라.

단, 빈 배열을 return해야한다면 [-1]을 return한다.

 

1 ≤ arr의 길이 ≤ 1,000,000

arr의 원소는 0 또는 1 입니다.

 

arr	result
[0, 1, 1, 1, 0]	[0, 1, 0]
[0, 1, 0, 1, 0]	[0, 1, 0, 1, 0]
[0, 1, 1, 0]	[-1]

** 풀이 방법

 

문제 설명 그대로 풀었다.

i의 초기값을 0으로 설정하고 i가 arr의 길이보다 작으면 다음을 반복한다.

- 만약 빈 배열이라면 arr[i]를 추가하고 i에 1을 더한다.

int i = 0;
while (i<arr.length) {
    if (list.size()==0) {
        list.add(arr[i]);
        i++;
    } 
    ...
}

 

- 원소가 있고, 마지막 원소가 arr[i]와 같으면 마지막 원소를 제거하고 i에 1을 더한다.

- 원소가 있는데 마지막 원소가 arr[i]와 다르면 맨 마지막에 arr[i]을 추가하고 i에 1을 더한다.

int i = 0;
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++;
        }
    }
}

 

단, 빈 배열을 return해야한다면 [-1]을 return한다.

if (list.size()==0) list.add(-1);

 

<> 전체 코드 </>

import java.util.*;
class Solution {
    public List<Integer> solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        int i = 0;
        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) list.add(-1);
        return list;
    }
}