目录
一、问题描述
二、解题思路
三、代码
四、复杂度分析
一、问题描述
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素。元素的顺序可能发生改变。然后返回 nums
中与 val
不同的元素的数量。
假设 nums
中不等于 val
的元素数量为 k
,要通过此题,您需要执行以下操作:
- 更改
nums
数组,使nums
的前k
个元素包含不等于val
的元素。nums
的其余元素和nums
的大小并不重要。 - 返回
k
。
用户评测:
评测机将使用以下代码测试您的解决方案:
int[] nums = [...]; // 输入数组 int val = ...; // 要移除的值 int[] expectedNums = [...]; // 长度正确的预期答案。// 它以不等于 val 的值排序。int k = removeElement(nums, val); // 调用你的实现assert k == expectedNums.length; sort(nums, 0, k); // 排序 nums 的前 k 个元素 for (int i = 0; i < actualLength; i++) {assert nums[i] == expectedNums[i]; }
如果所有的断言都通过,你的解决方案将会 通过。
二、解题思路
为了解决这个问题,我们可以使用 双指针 技巧。一个指针遍历整个数组,另一个指针记录不等于 val
的元素的位置。通过这样的方式,我们可以原地修改数组并将等于 val
的元素“移除”。
- 初始化指针:
- 使用一个指针
i
遍历整个数组,另一个指针k
用来记录下一个存储不等于val
的元素的位置。
- 使用一个指针
- 遍历数组:
- 遍历
nums
数组的每个元素,如果nums[i] != val
,将nums[i]
赋值给nums[k]
,并将k
向右移动一位。 - 如果
nums[i] == val
,跳过该元素。
- 遍历
- 结束条件:
- 当遍历完整个数组后,指针
k
就是不等于val
的元素的数量,并且nums
数组的前k
个位置已经被修改为不等于val
的元素。
- 当遍历完整个数组后,指针
三、代码
class Solution {public int removeElement(int[] nums, int val) {// 初始化指针 k 记录不等于 val 的元素位置int k = 0;// 遍历数组for (int i = 0; i < nums.length; i++) {// 如果当前元素不等于 valif (nums[i] != val) {// 将其放置在 k 的位置,并移动 knums[k] = nums[i];k++;}}// 返回不等于 val 的元素的数量 kreturn k;}
}
四、复杂度分析
时间复杂度:
- O(n):我们遍历了一遍数组,其中
n
是数组的长度。
空间复杂度:
- O(1):我们只使用了常数级别的额外空间,直接在原数组上进行修改。