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  咸与维新  阅读(79)  评论(0)    收藏  举报