优质博文:IT-BLOG-CN
一、题目
给你一个非严格递增排列的数组nums
,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。元素的相对顺序应该保持一致。然后返回nums
中唯一元素的个数。
考虑nums
的唯一元素的数量为k
,你需要做以下事情确保你的题解可以被通过:更改数组nums
,使nums
的前k
个元素包含唯一元素,并按照它们最初在nums
中出现的顺序排列。nums
的其余元素与nums
的大小不重要。返回k
。
判题标准: 系统会用下面的代码来测试你的题解
int[] nums = [...]; // 输入数组
int[] expectedNums = [...]; // 长度正确的期望答案int k = removeDuplicates(nums); // 调用assert k == expectedNums.length;
for (int i = 0; i < k; i++) {assert nums[i] == expectedNums[i];
}
如果所有断言都通过,那么您的题解将被通过。
示例 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]
解释:函数应该返回新的长度5
, 并且原数组nums
的前五个元素被修改为0, 1, 2, 3, 4
。不需要考虑数组中超出新长度后面的元素。
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums
已按非严格递增排列
二、代码
思路: 由于给定的数组nums
是有序的,因此对于任意i<j
,如果nums[i]=nums[j]
,则对任意i≤k≤j
,必有nums[i]=nums[k]=nums[j]
,即相等的元素在数组中的下标一定是连续的。利用数组有序的特点,可以通过双指针的方法删除重复元素。定义两个指针fast
和slow
分别为快指针和慢指针,快指针表示遍历数组到达的下标位置,慢指针表示当前元素的下标位置,初始时两个指针都指向下标0
和1
。假设数组nums
的长度为n
。将快指针fast
依次遍历从1
到n−1
的每个位置,对于每个位置,如果nums[fast]≠nums[slow]
,说明nums[fast]
和之前的元素都不同,因此将nums[fast]
的值复制到nums[slow+1]
,然后将slow
的值加1
,即指向下一个位置。遍历结束之后,从nums[0]
到nums[slow]
的每个元素都不相同且包含原数组中的每个不同的元素,因此新的长度即为slow+1
,返回slow+1
即可。
class Solution {public int removeDuplicates(int[] nums) {// 思想:定义快慢两个指针fast/slow,初始化fast = slow + 1,当 fast == slow 时,只移动fast,否则移动 fast和slowif (nums.length == 0) {return 0;}int slow = 0, fast = 1;while (fast < nums.length) {if (nums[slow] != nums[fast]) {++slow;nums[slow] = nums[fast];}++fast;}return slow + 1;}
}
时间复杂度: O(n)
其中n
是数组的长度。快指针和慢指针最多各移动n
次。
空间复杂度: O(1)
只需要使用常数的额外空间。
双指针: 首先注意数组是有序的,那么重复的元素一定会相邻。要求删除重复元素,实际上就是将不重复的元素移到数组的左侧。考虑用2
个指针,一个在前记作p
,一个在后记作q
,算法流程如下:
【1】比较p
和q
位置的元素是否相等。
【2】如果相等,q
后移1
位
【3】如果不相等,将q
位置的元素复制到p+1
位置上,p
后移一位,q
后移1
位
【4】重复上述过程,直到q
等于数组长度。
返回p + 1
,即为新数组长度。
public int removeDuplicates(int[] nums) {if(nums == null || nums.length == 0) return 0;int p = 0;int q = 1;while(q < nums.length){if(nums[p] != nums[q]){nums[p + 1] = nums[q];p++;}q++;}return p + 1;}
时间复杂度: O(n)
。
空间复杂度: O(1)
。
优化: 这是大部分题解都没有提出的,在这里提一下。
考虑如下数组:
此时数组中没有重复元素,按照上面的方法,每次比较时nums[p]
都不等于nums[q]
,因此就会将q
指向的元素原地复制一遍,这个操作其实是不必要的。
因此我们可以添加一个小判断,当q - p > 1
时,才进行复制。
public int removeDuplicates(int[] nums) {if(nums == null || nums.length == 0) return 0;int p = 0;int q = 1;while(q < nums.length){if(nums[p] != nums[q]){if(q - p > 1){nums[p + 1] = nums[q];}p++;}q++;}return p + 1;
}
时间复杂度: O(n)
。
空间复杂度: O(1)
。