26. 删除有序数组中的重复项
题目: 将一个升序排列的数组 nums 原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。不需要考虑数组中超出新长度后面的元素。元素的相对顺序应该保持一致。
题中:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
————————————————————————————————————————————
解题思路:
排好序的数组,重复的元素一定是相邻的,所以可以用双指针的方法,一个指针 i 准备接收新元素,另一个指针 j 用于判断它所指的元素是新的还是与 i-1 重复的。
举例: [1,2,2,3,3] --> [1,2,3]
nums: | 1 | 2 | 2 | 3 | 3 |
---|---|---|---|---|---|
指针 | i= 1, j=1 |
nums[j] != nums[i-1] ,新元素,nums[i]接收,i++,j++。
nums: | 1 | 2 | 2 | 3 | 3 |
---|---|---|---|---|---|
指针 | i=2, j=2 |
nums[j] == nums[i-1],重复元素,j++。
nums: | 1 | 2 | 2 | 3 | 3 |
---|---|---|---|---|---|
指针 | i=2 | j=3 |
nums[j] != nums[i-1] ,新元素,nums[i]接收,i++,j++。
nums: | 1 | 2 | 3 | 3 | 3 |
---|---|---|---|---|---|
指针 | i=3 | j=4 |
nums[j] == nums[i-1],重复元素,j++。
nums: | 1 | 2 | 3 | 3 | 3 |
---|---|---|---|---|---|
指针 | i=3 |
j=5结束,返回i。
代码:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int i = 1;
for (int j = 1; j < nums.size(); ++j)
{
if (nums[j] != nums[i-1]) //新元素赋值给nums[i]
{
nums[i] = nums[j];
i++;
}
}
return i; //不同元素个数为i
}
};
80. 删除有序数组中的重复项 II
题目: 将一个有序数组 nums 原地删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。不能使用额外的数组空间,即O(1) 额外空间。
题中:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
————————————————————————————————————————————
解题思路:
思想同上题,不同点在于判断nums[j]是不是出现超过2次,即判断 nums[j] 和 nums[i-2] 是否相等。如果数组长度不超过2直接返回。
举例: [1,1,1,2,2,2,3] —>[1,1,2,2,3]
nums: | 1 | 1 | 1 | 2 | 2 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i= 2, j=2 |
nums[j] == nums[i-2],超过2个,j++。
nums: | 1 | 1 | 1 | 2 | 2 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i= 2 | j=3 |
nums[j] != nums[i-2] ,不超过2个,nums[i]接收,i++,j++。
nums: | 1 | 1 | 2 | 2 | 2 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i=3 | j=4 |
nums[j] != nums[i-2] ,不超过2个,nums[i]接收,i++,j++。
nums: | 1 | 1 | 2 | 2 | 2 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i=4 | j=5 |
nums[j] == nums[i-2],超过2个,j++。
nums: | 1 | 1 | 2 | 2 | 2 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i=4 | j=6 |
nums[j] != nums[i-2] ,不超过2个,nums[i]接收,i++,j++。
nums: | 1 | 1 | 2 | 2 | 3 | 2 | 3 |
---|---|---|---|---|---|---|---|
指针 | i=5 |
j=7结束,返回i。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() <= 2)
return nums.size();
int i = 2;
for (int j = 2; j < nums.size(); ++j)
{
if (nums[j] != nums[i - 2])
{
nums[i] = nums[j];
i++;
}
}
return i;
}
};