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;
浙公网安备 33010602011771号