一、题目
二、解题思路
1、我的思路
因为题目中说“元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。”也就是说:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2
并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
所以我的想法是,我们要尽量避免在数组中删除元素这种事件复杂度高的操作,所以当元素值和val相同时,我们可以不必将它删除,而是将它和数组的最后一个元素交换位置,同时让数组长度减1
和上一题(上一篇博客)的想法一样:我们定义一个计数器count,来记录交换元素的次数,nums.length - count 即为新数组的长度
值得注意的是:在交换之后,我们不知道换到前面的数是否也等于val,所以还需执行i--
于是我写了如下代码,第一次提交就通过了,而且代码时间效率击败了100%的Java用户,给我整笑了
int count = 0;
for (int i = 0; i < nums.length - count; i++) {
if(nums[i] == val){
int temp = nums[i];
nums[i] = nums[nums.length - count -1];
nums[nums.length - count -1] = temp;
count++;
i--;
}
}
return nums.length - count;
2、官方题解
方法一:双指针
这题官方题解的思路和上一题(上一篇博客)基本一样
class Solution {
public int removeElement(int[] nums, int val) {
int n = nums.length;
int left = 0;
for (int right = 0; right < n; right++) {
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
}
作者:力扣官方题解
链接:https://leetcode.cn/problems/remove-element/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
方法二:双指针优化
class Solution {
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;
}
}
作者:力扣官方题解
链接:https://leetcode.cn/problems/remove-element/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。