[Java] 80. Remove Duplicates from Sorted Array II

문제 파악

nums 는 비내림차순 정수 배열이며, 최대 2개까지 중복될 수 있도록 나머지 중복은 제거해야한다.

그리고, 상대적인 위치를 유지해야한다.

주어진 조건에 맞게 중복을 제거한 원소의 개수 k를 반환해야한다.

nums 배열 외 다른 배열을 사용하면 안된다.

풀이

1️⃣ 포인터 방식

nums 배열만 사용해야하기 때문에, nums 배열의 원소들을 갱신해나가며 중복을 최대 2개까지 허용하면서 중복을 제거해야겠다고 생각했다.

먼저, 갱신할 수 있는 조건이 뭔지 생각해봤다.

원소가 달라지는 순간 갱신하고 포인터를 이동
원소가 같은 경우
1. 중복된 원소가 2개까지는 허용되므로, 2개까지는 갱신하고 포인터를 이동
2. 단, 3개부터는 갱신하지 않고 원소가 달라지는 순간까지 포인터를 움직일 수 없음

위 조건을 먼저 생각해보고나서, 코드로 구현을 해보았다.

class Solution {
    public int removeDuplicates(int[] nums) {
        int idx = 1;
        int cnt = 1;
        for(int i=1;i<nums.length;i++){
        	// 원소가 같은 경우
            if(nums[i-1]==nums[i]){
            	// 카운트 증가
                cnt++;
                
                // 2개까지는 허용하므로 갱신 후 포인터 증가
                if(cnt<=2){
                    nums[idx++] = nums[i];
                }
            }else{
            
            	//원소가 다른 경우 카운트를 1로 초기화 후, 갱신
                cnt=1;
                nums[idx++] = nums[i];
            }
        }

        return idx;
    }
}