(Python, Java) 프로그래머스 - 같은 숫자는 싫어

02 Jan 2022 in Algorithm on Programmers, Level1

[문제 링크]

Python 풀이

def solution(arr):
    answer = [arr[0]]

    for num in arr[1:]:
        if num == answer[-1]:
            continue
        else:
            answer.append(num)
    return answer

Java 풀이

import java.util.*;

public class Solution {
    public int[] solution(int[] arr) {
        LinkedList<Integer> answer = new LinkedList<>();
        answer.add(arr[0]);
        for (int i = 1; i < arr.length; i++) {
            if (answer.getLast() != arr[i]) {
                answer.add(arr[i]);
            }
        }

        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}