[Java] 27. Remove Element

문제 파악

정수 배열 nums와 val이 주어진다.

 

배열 nums 에서 val과 일치하는 원소를 모두 제거해야한다.

 

그리고 nums 배열에서 val과 일치하지 않는 원소의 개수를 반환해야한다.

 

일치하지 않는 원소의 개수가 k 일 때, nums의 k번 인덱스 까지는 val과 일치하지 않는 원소들로 채워넣어야 한다.

 

---

풀이

 

1️⃣ Arrays.sort() 사용

 

먼저, 문제 조건을 보니 nums 배열의 원소 최댓값은 50임을 알 수 있었다.

 

따라서, nums 배열에서 val과 일치하는 원소를 51로 모두 변환해두고

 

정렬을 수행하면, 배열의 뒷부분으로 밀려나 있을 것으로 예상했다.

 

import java.util.*;
class Solution {
    public int removeElement(int[] nums, int val) {
        int len = nums.length;
        int cnt = 0;
        
        for(int i=0;i<len;i++){
            if(nums[i]==val){
                nums[i] = 51;
                cnt++;
            }
        }

        Arrays.sort(nums);

        return len - cnt;
    }
}

 

결과

 

 

 

2️⃣ 포인터 방식

 

포인터 방식을 사용하면, 정렬을 수행하지 않을 수 있다.

 

nums에서 val과 일치하는 인덱스인 경우, 포인터를 멈추고

 

일치하지 않는 경우에만 값을 갱신하고 포인터를 이동시키면 된다.

 

import java.util.*;
class Solution {
    public int removeElement(int[] nums, int val) {
        int idx = 0;

        for(int i=0;i<nums.length;i++){
            if(nums[i]!=val){
                nums[idx++] = nums[i];
            }
        }

        return idx;
    }
}

 

 

결과

 

---

 

📖 회고

 

두번째 풀이 방식의 경우, 직관적으로 떠올리지 못했다.

 

배열의 길이가 한정적이라, 정렬을 사용해도 문제가 없었지만.

 

배열 문제를 풀 때, 대부분 O(N) 의 풀이 방법은 포인터를 활용하는 방법이니

 

한번쯤은 포인터 풀이 방법을 고민해봐야겠다.