题目
给定一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
思路
数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖
暴力解法
两层for循环,一个for循环遍历数组元素 ,第二个for循环更新数组。
class Solution {
public int removeElement(int[] nums, int val) {
int length = nums.length;
for (int i = 0; i < length; i++) {
if (nums[i] == val) { // 发现需要移除的元素,就将数组集体向前移动一位
for (int j = i + 1; j < length; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
length--; // 此时数组的大小-1
}
}
return length;
}
}
双指针法
通过两个指针在一个for循环下完成两个for循环的工作。
指针1:指向更新 新数组下标的位置
指针2:寻找新数组的元素 值,新数组就是不含有目标元素的数组
class Solution {
public int removeElement(int[] nums, int val) {
// 双指针
int Index1 = 0;//指向数组第一个元素
for (int Index2 = 0; Index2 < nums.length; Index2++) {//
if (nums[Index2] != val) {//只要当前元素不等于目标值
nums[Index1] = nums[Index2];//就将元素加入到新数组(实际上就是实现原地的覆盖修改)
Index1++;//更新完值后指针后移
}
}
return Index1;
}
}
这种方法并没有改变元素的相对位置
时间复杂度O(n)
, 空间复杂度O(1)