【题目描述】
给你一个 非严格递增排列 的数组 nums
,请你原地删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
考虑 nums
的唯一元素的数量为 k
,你需要做以下事情确保你的题解可以被通过:
一、更改数组 nums
,使 nums
的前 k
个元素包含唯一元素,并按照它们最初在 nums
中出现的顺序排列。nums
的其余元素与 nums
的大小不重要。
二、返回 k
。
【测试用例】
示例1:
输入:nums = [1,1,2]
输出:2,nums = [1,2]
解释:函数应该返回新的长度2,并且原数组nums的前两个元素被修改为1,2。不需要考虑数组中超出新长度后面的元素。
示例2:
输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5,nums = [0,1,2,3,4]
【思路分析】
这里用的思想类似于空间换时间,用一个新的数组expectedNums去存储去重之后的数组。具体的步骤就是,先把nums[0]存入expectedNums[0],然后依次遍历nums中后面的元素:
如果nums[i]等于expectedNums中的最后一个元素,说明出现了重复元素,继续判断nums[i+1];
如果nums[i]不等于expectedNums中的最后一个元素,说明元素没有重复,将num[i]存入数组expectedNums的末尾。
注意:这里的做法是我最开始的做法,后来在去看题解的时候看见有的评论说的题目要求了“原地删除重复出现的元素”,所以这种做法虽然能AC,但是应该算是捡漏了。正确的做法不能创建额外的数组空间,要在原数组nums中修改。具体做法我写在了【思路更新】中。
【参考代码】
C实现
#include <stdio.h>
#include <stdlib.h>
//easy-26-删除有序数组中的重复项
int removeDuplicates(int* nums, int numsSize);
int main(){
int nums[] = {1,1,2};
int numsSize = sizeof(nums)/sizeof(nums[0]);
int k = removeDuplicates(nums,numsSize);
int i;
printf("%d\n",k);
for(i=0;i<k;++i){
printf("%d ",nums[i]);
}
return 0;
}
int removeDuplicates(int* nums, int numsSize) {
int* expectedNums = (int*)malloc(sizeof(int)*numsSize);
int i,k=0;
expectedNums[0] = nums[0];
for(i=1;i<numsSize;++i){
if(nums[i] != expectedNums[k]){
expectedNums[++k] = nums[i];
}
}
for(i=0;i<k+1;++i){
nums[i] = expectedNums[i];
}
return k+1;
}
C++实现
#include <iostream>
#include <vector>
using namespace std;
//easy-26-删除有序数组中的重复项
class Solution {
public:
int removeDuplicates(vector<int>& nums);
};
int Solution::removeDuplicates(vector<int>& nums){
vector<int> expectedNums;
int i,k=0;
expectedNums.push_back(nums[0]);
for(i=1;i<nums.size();++i){
if(nums[i] != expectedNums[k]){
expectedNums.push_back(nums[i]);
k++;
}
}
for(i=0;i<k+1;++i){
nums[i] = expectedNums[i];
}
return k+1;
}
int main(){
vector<int> nums = {1,1,2};
Solution sol;
int k = sol.removeDuplicates(nums);
cout<<k<<endl;
int i;
for(i=0;i<k;++i){
cout<<nums[i]<<" ";
}
return 0;
}
【思路更新】
不能创建额外空间,既要操作数组nums的遍历,又要操作数组nums元素的修改,很容易想到双指针(说是指针,实则就是两个记录数组元素位置的变量,用于访问某个位置的元素)。
一个指针pos1从nums[0]开始,用于记录需要修改元素时的位置(相当于之前的思路中数组expectedNums中的位置);
另一个指针pos2从nums[1]开始向后遍历,判断当前指向的元素nums[pos1]与上一个指针指向元素nums[pos2]的关系:
如果nums[pos1] = nums[pos2],说明出现了重复的元素,同样的做法,pos2后移,继续比较;
如果nums[pos1] != nums[pos2],说明没有重复,pos1后移,pos2后移,继续比较。
当pos2移动到nums的末尾位置时循环结束。
ps:题目中规定了“
nums
的其余元素与nums
的大小不重要”,变相的减轻了题目难度,函数返回值是不重复的元素个数,在这里实则就是pos2+1,最后输出的时候只需要输出nums中的前pos2+1个元素即可。
【参考代码】
这里只更新了核心函数removeDuplicates中的代码,因为其他的部分都与上面的代码一样。(其实C和C++两种实现内部的代码也是几乎一样的哈哈哈哈)
C实现
int removeDuplicates(int* nums, int numsSize) {
int pos1=0, pos2=1;
while(pos2<numsSize){
if(nums[pos2] == nums[pos1]){
pos2++;
}else{
nums[++pos1] = nums[pos2++];
}
}
return pos1+1;
}
C++实现
int Solution::removeDuplicates(vector<int>& nums){
int pos1=0, pos2=1;
while(pos2 < nums.size()){
if(nums[pos2] == nums[pos1]){
pos2++;
}else{
nums[++pos1] = nums[pos2++];
}
}
return pos1+1;
}