给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 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 。不需要考虑数组中超出新长度后面的元素。
思路
双指针法是数组、链表的操作中非常常见的。
这题的前提是:升序排序的数组 nums
。意味着存在重复的元素是相邻的。
我们定义两个指针分别指向数组下标为 0、1
位置,记作指针 p 、q
。
当 nums[p] == nums[q]
时,p
指针后移一位。
当 nums[p] != nums[q]
时,将 p
指针指向的值赋值给 q
指针后一个值。同时 q+1
如nums[p+1] = nums[q]
如此循环直到 q
大于数组大小。
代码
暴力写法
class Solution {
public int removeDuplicates(int[] nums) {
// 利用map 的无重复元素的特点进行排除
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
int size = nums.length;
for (int i = 0; i < size; i++) {
if ( map.get(nums[i]) != null && map.get(nums[i]) == nums[i]) {
for(int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--;
size--;
}
map.put(nums[i],nums[i]);
}
return size;
}
}
双指针法
class Solution {
public int removeDuplicates(int[] nums) {
// 双指针法
// 定义算两个指针
int p = 0, 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;
}
}