给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
例1:
给定 nums = [0,0,1,1,1,2,2,3,3,4], 函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0,1,2,3,4 你不需要考虑数组中超出新长度后面的元素。
解题思路:
很容易想到的就是利用两个指针,其中快指针负责遍历整个数组(容器),慢指针只负责累加非重复项。
(1)最开始的代码
int removeDuplicates(vector<int>& nums) {
// Method 1 Time costs 40ms ranking 60%
int n = (int)nums.size();
if(nums.empty() || n == 0)
return 0;
if(n == 1)
return 1;
int size = 0;
int temp = nums[0];
for(int i = 0; i < n; ) {
while(temp == nums[i])
i++;
temp = nums[i];
if(i < n)
nums[++size] = nums[i++];
}
return size+1;
}
这段代码效率不高,还多使用了一个temp临时变量去做比较,其实完全可以用慢指针指向的元素进行比较;
(2)优化后的代码
int removeDuplicates(vector<int>& nums) {
// Method 2 Time costs 20ms
int size = !nums.empty();
for(int n : nums) {
if(n > nums[size-1])
nums[size++] = n;
}
return size;
}
其中for循环使用了c++的新标准,看着很简洁,由于我们要进行处理的数组(容器)是顺序的,因此我们直接比大小。
(3)提交后发现还有极小一部分运行测试仅用了16ms,看了其源代码后,自己重新写了一边并多次提交,发现测试结果不固定,并不是算法又有了提高,而是提供的测试数据是不一样的,因此运行时间大概在一个范围 16ms~40ms
int removeDuplicates(vector<int>& nums) {
// Method 3 Time costs 16ms ~ 40ms according to test data
int size = 0, i = 0;
while(i < nums.size()) {
while(size && nums[i] == nums[size-1])
i++;
if(i < nums.size())
nums[size++] = nums[i++];
}
return size;
}
最后给出在编写代码时的主函数,方便进行测试。
int main(int argc, const char * argv[]) {
// enter any number of integers
vector<int> nums;
int n;
while(cin >> n)
nums.push_back(n);
int size = removeDuplicates(nums);
cout << "删除重复数之后的数组(容器)长度:" << size << endl;
cout << "遍历结果为:" << endl;
for(n = 0; n < size; n++){
cout << nums[n] << " ";
}
cout << endl;
return 0;
}
输入任意数量整数,并以随意一个字符结束输入。例如:
1 2 2 3 3 4 5 6 7 7q
删除重复数之后的数组(容器)长度:7
遍历结果为:
1 2 3 4 5 6 7
Program ended with exit code: 0