[Lintcode] Remove Duplicates from Sorted Array II

最新推荐文章于 2023-09-22 13:35:40 发布

原创最新推荐文章于 2023-09-22 13:35:40 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种算法解决方案，该方案允许在一个已排序的数组中移除多余的重复元素，但允许每个元素最多出现两次。例如，对于数组 [1,1,1,2,2,3]，经过处理后，数组将变为 [1,1,2,2,3]，长度为5。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array A = [1,1,1,2,2,3],

Your function should return length = 5, and A is now [1,1,2,2,3].

与第一道题大体一致，只需额外引入一个变量标记是否是第二次出现即可

public class Solution {
    /**
     * @param A: a array of integers
     * @return : return an integer
     */
    public int removeDuplicates(int[] nums) {
        if(nums.length == 0) return 0;
        int index = 0;
        boolean twice = false;
        
        for(int i = 1; i < nums.length; i++) {
            if(nums[index] != nums[i] || !twice) {
                if(nums[index] == nums[i]) {
                    twice = true;
                }
                else {
                    twice = false;
                }
                nums[index + 1] = nums[i];
                index++;
            }
        }
        return index + 1;
    }
}