给一个数组和一个值,移除数组中所有该元素并返回新的长度。不可以分配新的空间。
Example 1:
nums = [3,2,2,3], val = 3,
返回长度为2,数组为[2,2]
Example 2:
nums = [0,1,2,2,3,0,4,2], val = 2,
返回长度为5,数组为[0,1,3,0,4]
根据题意,移除数组中值为val的元素后的数组长度,保证算法的空间复杂度是O(1)。
假设数组为nums,首先我们肯定是需要遍历数组的,使用两个指针i和j,指向数组第一个元素。接下来需要判断所遍历到的元素num[j]是否等于val
- 如果不等于val,i和j都向后移。
- 如果等于val,j往后移,找到最近的不等于val的元素,让nums[i] = nums[j]。
在第二种情况下,i的大小就是移除数组中值为val的元素后的数组长度。
代码如下:
/*第一种情况*/
class Solution {
public int removeElement(int[] nums, int val) {
int i = 0;
for(int j = 0; j < nums.length; j++){
if(nums[j] != val){//如果不等于val,i和j都向后移。
i++;
}
}
return i;
}
}
/*第二种情况*/
class Solution {
public int removeElement(int[] nums, int val) {
int i = 0;
for(int j = 0; j < nums.length; j++){
if(nums[j] != val){//如果不等于val,i和j都向后移。
i++;
nums[i] = nums[j];//此时j为最近的不等于val的元素下标。
}
}
return i;
}
}