给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
1 2 3 4 5 6 7 8
| int len = removeElement(nums, val);
for (int i = 0; i < len; i++) { print(nums[i]); }
|
示例 1:
1 2 3
| 输入:nums = , val = 3 输出:2, nums = 解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = 或 nums = ,也会被视作正确答案。
|
示例 2:
1 2 3
| 输入:nums = [0,1,2,2,3,0,4,2], val = 2 输出:5, nums = [0,1,4,0,3] 解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
|
双指针法
slow指针负责把不需要删除的元素复制到原来的数组
1 2 3 4 5 6 7 8 9 10 11
| class Solution { public int removeElement(int[] nums, int val) { int slow = 0; for(int i = 0 ; i < nums.length; i++){ if(nums[i] != val){ nums[slow++] = nums[i]; } } return slow; } }
|
时间复杂度:$O(n)$
给你一个 非严格递增排列 的数组 nums
,请你** 原地** 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
考虑 nums
的唯一元素的数量为 k
,你需要做以下事情确保你的题解可以被通过:
- 更改数组
nums
,使 nums
的前 k
个元素包含唯一元素,并按照它们最初在 nums
中出现的顺序排列。nums
的其余元素与 nums
的大小不重要。
- 返回
k
。
判题标准:
系统会用下面的代码来测试你的题解:
1 2 3 4 5 6 7 8 9
| int[] nums = [...] int[] expectedNums = [...]
int k = removeDuplicates(nums)
assert k == expectedNums.length for (int i = 0 assert nums[i] == expectedNums[i] }
|
如果所有断言都通过,那么您的题解将被 通过。
示例 1:
1 2 3
| 输入:nums = 输出:2, nums = 解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。
|
示例 2:
1 2 3
| 输入:nums = [0,0,1,1,1,2,2,3,3,4] 输出:5, nums = [0,1,2,3,4] 解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
|
slow指向待插入位置,因为需要找到一个与nums[0]不相同的元素插入,故slow=1;
j遍历数组,找到与slow前一个元素不相同的插入
1 2 3 4 5 6 7 8 9 10 11
| class Solution { public int removeDuplicates(int[] nums) { int slow = 1; for(int i = 1 ; i < nums.length; i++){ if(nums[i] != nums[slow-1]){ nums[slow ++] = nums[i]; } } return slow; } }
|
移除元素使得重复元素次数为2
如果元素可重复多次
可用count进行统计,count表示已经重复的次数
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
| class Solution { public int removeDuplicates(int[] nums) { int slow = 1; int count = 1; for(int i = 1; i < nums.length; i++){ if(nums[i] != nums[slow-1]){ nums[slow ++] = nums[i]; count = 1; }else{ if(count < 2){ count ++; nums[slow ++] = nums[i]; } } } return slow; } }
|