一. 题目解读,为什么返回数值是整数,但输出的答案是数组呢?请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的.传入的是数组的地址.
二. 这题是学习题,学习一下如何解决此类问题.参考大神的直接遍历一次即可.
作者:luo-ben-zhu-xiao-man-tou
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/solution/zhi-jie-bian-li-yi-ci-ji-ke-by-luo-ben-zhu-xiao-ma/
1. 原地删除肯定是双指针,一个指向遍历的元素,一个指向可以写入的位置,后者的大小是小于等于前者的,关键在于题目条件的转化,如何实现限制最多两次的重复出现。
2. 我们先不考虑边界情况,只考虑中间的情况,假设当前遍历位置为i,写指针的可写入位置为current+1,对于i处的值,其写入的条件是重复小于等于2次,我们考虑已经写入的最后两位current和current-1,这两个位置的情况有两个,相等和不相等,首先考虑相等的情况,此时若i处的值和current-1或者说current处的值相同,那么,i处的值肯定不能加入;然后考虑不相等的情况,即current-1和current处值不相等,那么i处的值无论为什么,都满足题意的,即可以加入,综上所述,当i处的值与current-1处的值不相等时,i处的值可以加入,其他情况均不能加入。
3. 接着考虑边界情况,我们只需要考虑开始即可,开始时,前两个值无论等还是不等,都要原封不动的挪到新数组里,由于新数组就是在原数组上进行修改的,因此前两位直接不动即可,只需要修改遍历指针和写入指针就行。
4. 以上算法只需要进行一次遍历即可,时间复杂度O(n),空间复杂度O(1)。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
//边界条件,小于等于1长度的直接返回即可.
if (nums.size() <= 1)
return nums.size();
//因为每个元素最多两次,则
//如果nums[i]不等于nums[current-1]
//则它一定可以加入current的后一个索引.
int current = 1;
//前两个一定加入,从2开始.
for (int i = 2; i < nums.size(); i++) {
if (nums[i] != nums[current - 1])
nums[++current] = nums[i];
}
return current + 1;
}
};
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() <= 1)
return nums.size();
int current = 1;
for (int i = 2; i < nums.size(); i++) {
//正常想法,只有当i和current还有current-1的值相等时,才跳过.
//我前面取个反!,则需要将i加入++current.
if (!(nums[i] == nums[current] && nums[i] == nums[current - 1]))
nums[++current] = nums[i];
}
return current + 1;
}
};
5. 拓展思维,可以引入一个通用模板,如果题目改成最多重复k元素代码如下:(因为数组已经排好序)
class Solution:
def removeDuplicates(self, nums, k):
i = 0
for n in nums:
if i < k or n != nums[i-k]:
nums[i] = n
i += 1
return i