问题
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
首尾指针法
main方法:
public class Exercise2_1 {
public static void main(String[] args) {
int[] arr = {5,6,44,5,6,7,6,6};//在函数中会对其进行直接移动
Solution2_1 solution2_1 = new Solution2_1();
int num = solution2_1.removeElement(arr,6);
for (int i = 0; i < num; i++) {
System.out.print(arr[i]+ " ");
}
}
}
执行方法:
右指针保证是数组长度(而不是为了为了保证右指针是数组下标设为 数组长度-1),保证下面while循环次数足够
class Solution2_1 {
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;
}
}