题目描述
给你一个非严格递增排序的数组nums,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。元素的相对顺序应该保持一致,然后返回nums中唯一元素的个数。
考虑nums的唯一元素的数量为k,你需要做以下事情确保你的题解可以被通过:
1.更改数组nums,使nums的前k个元素包含唯一元素,并按照它们最初在nums中出现的顺序排列。nums的其余元素与nums的大小不重要。
2.返回k。
示例
输入:nums={0,0,1,1,1,2,2,3,3,4}
输出:5,nums={0,1,2,3,4}
解释:函数应该返回新的长度5,并且原数组nums的前五个元素被修改为0,1,2,3,4。不需要考虑数组中超出新长度后面的元素。
题解
首先先判断数组的长度是否为0,若为0,则直接返回;若不为0,则进行以下步骤。
int removeDuplicates(int* nums, int numSize)
{
if(numSize == 0)
return 0;
}
创建3个变量prev、cur和dst,使nums[prev]和nums[dst]表示第一个0,nums[cur]表示第二个0,即prev = dst = 0,cur = 1。
若nums[prev]与nums[cur]相等,则dst不动,prev和cur同时向后走一步;若nums[prev]与nums[cur]不相等,则用nums[prev]替换掉nums[cur],接着prev、cur和dst同时向后走一步。
int prev = 0, cur = 1, dst = 0;
while(cur < numsSize)
{
if(nums[prev] != nums[cur])
{
nums[dst] = nums[prev];
prev++;
cur++;
dst++;
}
else
{
prev++;
cur++;
}
}
最后,直接用nums[prev]替换掉nums[dst],prev和dst再同时往后走,并返回dst(此时dst=5)。
nums[dst] = nums[prev];
dst++;
prev++;
return dst;
整体代码如下: