80. 删除有序数组中的重复项 II(LeetCode)

题目条件


给你一个有序数组 nums(升序) ,请你 原地 删除重复出现的元素,使每个元素 最多出现两次 ,返回删除后数组的新长度。

不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

输入:nums = [1,1,1,2,2,3]

输出:5, nums = [1,1,2,2,3]

解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 不需要考虑数组中超出新长度后面的元素。

条件分析


1.数组为升序
2.原地删除,考虑元素交换和赋值

解题思路


1.定义下标k,[0,k]之间的元素最多只出现两次;

2.为了确保最多出现2次,使用额外的变量count记录当前k对应元素出现的次数,count初始值为1

3.使用下标i遍历数组

(1)如果i所在元素和k所在元素相等且++count后小于2,此时k++,并移动i到k处

(2)当i所在元素和k不相等时,由于数组是升序排列,说明i和[0,k]之间的所有元素都不相等,此时k++,并移动i到k处,并且设置count=1

4.k为数组下标,遍历完成后,返回k+1,才是数组长度

编码如下

if (nums.length == 0) {
		return 0;
	}
	int k = 0;
	int count = 1;
	for (int i=1; i<nums.length; i++) {
		if (nums[i] != nums[k] || (nums[i] == nums[k] && ++count <= 2)) {
            if (nums[i] != nums[k]) {
				count = 1;
			}
			k++;
			if (k != i) {
				nums[k] = nums[i];
			}
		}
	}
	return k+1;
posted @ 2021-07-07 20:30  咸与维新  阅读(67)  评论(0)    收藏  举报