[프로그래머스/JAVA] 같은 숫자는 싫어

https://programmers.co.kr/learn/courses/30/lessons/12906

코딩테스트 연습 - 같은 숫자는 싫어

 

 

 

문제 설명이 길지만 배열의 중복을 제거하는 문제이다.

ArrayList를 이용해서 원소를 추가하는 방식으로 중복을 제거해서 풀었다.

 

맨 처음에 풀었을 때는 num이라는 변수없이 if(!list.contains(arr[i])) ~ 이런식으로 풀었더니 계속 틀렸었다.

 

문제에서 원소의 크기는 0~9까지라고 했으니까 절대 원소가 될 일 없는 10으로 변수 num을 초기화 한다.

 

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        
        ArrayList<Integer> list = new ArrayList<Integer>();
        
        int num = 10;
        
        for(int i=0; i<arr.length; i++){
            if(num != arr[i]){
                list.add(arr[i]);
                num = arr[i];
            }           
        }
        
        //answer배열에 답 저장
        int[] answer = new int[list.size()];       
        for(int i=0; i<answer.length; i++){
            answer[i] = list.get(i);
        }

        return answer;
    }
}

 

 

• 2024.10.13 Stack 으로 다시 풀어봄

1. Stack을 생성

2. Stack이 비어있거나 가장 위에 있는 값(peek())이 입력받은 배열값과 다르면 push -> 이렇게 하면 중복없이 Stack에 값이 쌓일 것

3. Stack 사이즈와 같은 answer 리턴 배열을 생성

4. pop은 Stack의 가장 위부터 실행되므로 반복문을 거꾸로 돌리면서 answer 배열에 값을 넣음

 

 

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        Stack<Integer> stack = new Stack<>();
        
        for(int i=0; i<arr.length; i++){
            if(stack.isEmpty() || stack.peek()!=arr[i]) {                
                stack.push(arr[i]);
            }
        }
        //stack 사이즈 answer 배열 생성
        int[] answer = new int[stack.size()];
        for(int i=stack.size()-1; i>=0; i--){
            answer[i] = stack.pop();
        }
        return answer;
    }
}