给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3], 函数应返回新长度 length =5
, 并且原数组的前五个元素被修改为1, 1, 2, 2,
3 。 你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,1,2,3,3], 函数应返回新长度 length =7
, 并且原数组的前五个元素被修改为0
, 0, 1, 1, 2, 3, 3 。 你不需要考虑数组中超出新长度后面的元素。
解题思路:
交换元素法,这个题跟删除重复元素的题一样,都是排序数组,遇到不一样的就往前放,只不过,这个可以有两个重复而已。
class Solution { public: int removeDuplicates(vector<int>& nums) { if (int(nums.size()) == 0) return 0; int pos = 0, sgn = 3, temp = nums[0]; for (int i = 1; i <= int(nums.size()); i++) { if (temp == nums[i - 1]) sgn--; else { sgn = 2; temp = nums[i - 1]; } if (sgn > 0) swap(nums[i - 1], nums[pos++]); } nums.erase(nums.begin() + pos, nums.end()); return int(nums.size()); } }; |