删除有序数组中的重复项 II（计数法，容易理解）

陈小桔

已于 2024-09-15 15:07:01 修改

阅读量179

点赞数 2

文章标签：算法数据结构 java

于 2024-09-15 14:56:30 首次发布

本文链接：https://blog.csdn.net/2401_84049789/article/details/142284217

版权

class Solution {
    public int removeDuplicates(int[] nums) {
        if(nums.length <= 2){
            return nums.length;
        }
        int cnt = 1, end = 1;
        for(int i = 1; i < nums.length; i++){
            if(nums[i] == nums[i - 1]){
                cnt++;
                if(cnt <= 2){
                    nums[end++] = nums[i];
                }
            }else{
                cnt = 1;
                nums[end++] = nums[i];
            }
        }
        return end;
    }
}

改成模板:

class Solution {
    public int removeDuplicates(int[] nums, int k) {
        if(nums.length <= k){
            return nums.length;
        }
        int cnt = 1, end = 1;
        for(int i = 1; i < nums.length; i++){
            if(nums[i] == nums[i - 1]){
                cnt++;
                if(cnt <= k){
                    nums[end++] = nums[i];
                }
            }else{
                cnt = 1;
                nums[end++] = nums[i];
            }
        }
        return end;
    }
}