题目
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-element
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我的代码
我的代码一直没通过,也没有存,这里没法贴了,先放上官方解,然后再说我遇到的问题。
官方代码
class Solution {
public int removeElement(int[] nums, int val) {
int i = 0;
for (int j = 0; j < nums.length; j++) {
if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}
return i;
}
}
首先,官方的代码可以说是简洁且优雅了。
我和官方的思路差不多,都是双指针来控制,主要再换元上。
我对题目的理解是,最后只检测返回的长度的长度,但实际上,是“移除”,也就是数组的长度是切切实实的减少了。
我用
int temp=nums[head];
nums[head] = nums[tail];
nums[tail]=temp;
这样并不是移除,所以不对。
参考了官方答案后,我改成
nums[head] = nums[tail];
奇怪的事情在这里发生了,讲道理这个替换方法与官方并没有什么不同,可是我的数组长度并不会减少,元素并没有删除,十分费解。
问题解决
经过一个晚上的冷静,我冷静了。
其实,虽然我和答案都是双指针,但是答案是快慢针,我是收尾针,这种情况下快慢针显然更为理想。首尾针会遇到许多问题,下面举例说明:
1.返回长度的计算
如[3,2,2,3]移除3,如果是首尾针,每次交换元素后ans–;这样的话ans=3,实则应为2
2.单元素输入问题
[2]移除1和移除2,各自都是一个问题。
总之,快慢针是个好东西,脑子也是。