题目如下:
解题思路:
本题刚开始的第一想法就是构建同一个空数组,再建立一个索引表(hash表),遍历每个元素并记录元素的访问次数,对不超过两次的元素则加入数组。但是题目明确要求不使用额外的数组空间,在O(1)的额外空间下完成。
因此看了一下评论区的思路,发现只需要遍历一次,并维护一个结果集数组的长度即可。在遍历的过程中,当结果集数组长度小于2或者当前元素不等于当前结果集数组中倒数第二个元素时,就将当前元素加入结果集,并增加结果集长度即可(对于不在结果集长度范围内的元素题目明确说明可以不用考虑)。
代码如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int res_len = 0; //用于记录新数组的长度
for(int i = 0; i < nums.size(); i++) {
//结果集长度小于2或者当前元素不等于当前结果集中倒数第二个元素,则将当前元素加入结果集
if(res_len < 2 || nums[i] != nums[res_len - 2]){
nums[res_len] = nums[i];
res_len++;
}
}
return res_len;
}
};