题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
解题思路
采用双指针方式处理。
- 初始指针 i, j 位置为 0
- 指针 j 先向右移动
- 比较指针 i,j 元素是否相同,若指针i,j元素相同则指针 j 继续向右移动;反之将指针 j 元素复制到指针 i + 1 处元素
- 当指针 j 移动到数组末尾时则停止
其流程如下图所示:
实现
public static int solution (int[] nums) {
int i = 0, j = 0;
while (true) {
// 指针 j 向右移动
j++;
// 指针 j 移动到数组末尾则退出,说明数组元素都判断了去重
if (j >= nums.length) {
break;
}
if (nums[j] == nums[i]) {
// 指针 i, j 元素相同,说明重复元素;
// 指针 j 继续向右移动
continue;
} else {
// 指针 i, j 元素不相同;则将指针 j 元素复制到 指针 i 后一位,这样就保证指针 i 后元素不重复
nums[i + 1] = nums[j];
// 指针 i 向右移动,继续处理
i++;
}
}
return i + 1;
}