Leetcode第80题，去除有序数组中的元素，允许重复k次

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
/**Question 80. 去掉排序数组中的重复元素
 * 给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。
 *
 * 不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

public class Solution80 {
    public static int removeDuplicates(int[] nums, int k) {
        if (nums.length == 0) return 0;
        if (nums.length < k) return nums.length;//如果允许重复的元素为k次，那么<k
        // 长度的数组都可以被直接返回，由于数组是从0开始计算下标的，所以没有==

        int index = k - 1; //这里的index定义为非重复元素的右边界，也就是[0...k)--->[0...index]
        int n = nums.length;    //数组的长度，进行到这一步，已经可以知道n >= k

        for (int j = 2; j < n; j++) //j是循环指针，用来遍历数组中[k...n)位置的所有元素
            if (nums[j] != nums[index - k + 1]) //判断j指向的元素是否在非重复元素集合内
                // 因为是有序数组，同时允许重复k次，所以只需要判断j所指向的元素与右边界往前数第k
                // 个元素是否相同，如果不相同，说明这个元素的可重复容量还未达到限定，可以将其加入。
                // 右边界为index，往前第k的位置为 index - k + 1这个位置，初始位置是0，但随着右边界的增大，左边界
                nums[++index] = nums[j];    //符合要求后，非重复元素增加一个新的元素，右边界+1

        return index + 1;
    }

}