给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
0 <= nums.length <= 3 * 10^4
0 <= nums[i] <= 10^4
nums 已按升序排列
示例 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 。不需要考虑数组中超出新长度后面的元素。
思路分析:
这个问题并不难,相信大家一定可以独立做出来。这里给出两种写法,两种写法里 i 都是用于遍历数组的下标,而 j 的含义不同,请大家分别比较两者代码的差异。
定义 1:j 指向将要赋值的元素的下标,因此我们定义区间 [0…j) (注意是左闭右开区间,j 不能取到)里没有重复元素。pre 永远指向最近的第 1 个不重复的数字;
-
初始化:为了保证区间 [0…j) (注意这里是左闭右开区间)里没有重复元素。初始的时候,i = 0 ,下标为 0 的位置只有一个数,这表示区间 [0…0] 没有重复数字,即区间 [0…1) 没有重复数字,因此 j 初始化的时候需要等于 1;pre = nums[ 0 ]。
-
保持:遇到和 pre 指向的数字相等元素,i++ ,直接下一个元素,如果 nums[ i ] != pre ,表示程序看到了第 1 个不重复的数字,此时需要赋值 nums[ j ] = nums[ i ] 和 pre = nums[ j ] ,然后让 j++ 指向下一个需要赋值的下标;
-
终止:循环结束时 i = len ,程序看完了输入数组的所有元素,此时区间 [0…j) 里没有重复元素,它的长度为 j ,返回 j。
参考代码 1:
public class Main {
public static int removeDuplicates(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int pre = nums[0];
int j = 1;
for (int i = 1; i < nums.length; i++) {
if (nums[i] != pre) {
nums[j] = nums[i];
pre = nums[j];
j++;
}
}
return j;
}
public static void main(String[] args) {
int[] nums = new int[] {0, 0, 1, 1, 1, 2, 2, 3, 3, 4};
int[] arr = null;
int len = removeDuplicates(nums);
// int len = removeDuplicates(arr);
for (int i = 0; i < len; i++) {
System.out.print(nums[i] + " ");
}
}
}
定义 2:我们定义:区间 [0…j] (注意这里是左闭右闭区间)没有出现重复元素,此时变量 j 是上一轮找到的第 1 次出现的元素的下标。因此不需要像「定义 1」一样给 pre 定义。循环变量 i 看到的数值 nums[i] 永远和变量 j 看到的数值 nums[j] 进行比较。 大家可以对照上面的「定义 1」自行写出「初始化」「保持」「终止」的逻辑。
参考代码 2:
public class Main {
public static int removeDuplicates2(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int j = 0;
for (int i = 1; i < nums.length; i++) {
if (nums[i] != nums[j]) {
nums[++j] = nums[i];
}
}
return ++j;
}
public static void main(String[] args) {
int[] nums = new int[] {0, 0, 1, 1, 1, 2, 2, 3, 3, 4};
int[] arr = null;
int len = removeDuplicates2(nums);
// int len = removeDuplicates(arr);
for (int i = 0; i < len; i++) {
System.out.print(nums[i] + " ");
}
}
}
复杂度分析:
时间复杂度:O(N),这里数组的长度是 N, i 和 j 分别最多走 N 步;
空间复杂度:O(1)。