题目描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
示例:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
动图演示
方式一:将右指针遍历到的不用删除的元素赋值到左指针位置
示例1:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
利用双指针原理,用right指针找到不等于val的元素,替换掉left指针所指位置元素。
循环结束后,[0,left)索引区域的元素都不等于left,left为移除val后的数组新长度。
方式二:用右指针的前一位替换掉左指针遍历到的要删除元素
示例2:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
总结:方式一将要保留元素赋值数组前面;方式二避免了需要保留的元素的重复赋值操作,只用后面元素替换要删除的元素。
代码
//方式一
lass Solution {
public int removeElement(int[] nums, int val) {
int n = nums.length;
int left = 0;
for (int right = 0; right < n; right++) {
//将右指针遍历到的不用删除的元素赋值到左指针位置
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
}
//方式二
class Solution {
public int removeElement(int[] nums, int val) {
int left = 0;
int right = nums.length; // *
while (left < right) { // *
if (nums[left] == val) {
//用右指针的前一位替换掉左指针遍历到的要删除元素
nums[left] = nums[right - 1]; // *
right--;
} else {
left++;
}
}
return left;
}
}
//right=nums.length-1 时写法
class Solution {
public static int removeElement(int[] nums, int val){
int left = 0;
int right = nums.length-1; // *
while (left<=right){ // *
if (nums[left]==val){
nums[left] = nums[right]; // *
right--;
}else {
left++;
}
}
return left;
}
}