对一个整数数组 nums 进行处理,以删除超过两次重复的数字,返回新数组的长度,并输出处理后的数组。
输入处理:
- 从控制台读取一行输入的整数(用逗号分隔的字符串形式)。 将字符串转化为整数,存入 nums 数组中。
核心算法(函数 removeDuplicates):
- 保证数组中每个数字最多只出现两次。
- 使用双指针技巧,其中:
k 代表当前符合要求的数组长度(也是插入位置的索引)。
i 遍历数组。 - 判断条件: 如果当前数字 nums[i] 不等于 nums[k-2](即确保最多两次重复),将数字添加到结果数组中。
- 返回值为修改后的数组长度 k
输出结果:
#include <iostream>
#include <vector>
#include <sstream>
#include <string>
using namespace std;
int removeDuplicates(vector<int>& nums){
if(nums.size() <= 2){
return nums.size();
}
int k = 2;
for(int i=2; i<nums.size(); i++){
if(nums[i] != nums[k-2]){
nums[i] = nums[k];
}
++k;
}
return k;
}
int main() {
string str;
vector<int> nums;
getline(cin,str);
stringstream ss(str);
string temp;
while(getline(ss, temp, ',')){
nums.push_back(stoi(temp));
}
int newLength = removeDuplicates(nums);
cout << newLength << endl;
cout << "nums = [";
for(int i=1; i<newLength; i++){
cout << nums[i] << " " ;
}
cout << "]" << endl;
return 0;
}