移除元素
27. 移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2,,]
解释:你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
解法1 一坨翔
用i记录数组中与val不相等的元素个数,边界条件过于复杂
完全根据错误测试用例对算法缝缝补补
class Solution {
public int removeElement(int[] nums, int val) {
// 双指针
// 1、i从从左往右扫描,j从右往左扫描。
// 2、i遇到val停止,j遇到非val停止,交换值,各进一步。
// 3、重复以上步骤,直到i > j
int len = nums.length;
int i = 0, j = len - 1;
if (i == j && nums[i] == val) return 0;
while (i < j) {
while (nums[i] != val) {
i++;
if (i == len) return i;
}
while (nums[j] == val) {
j--;
if (j == -1) return 0;
}
if (i > j) return i;
nums[i] = nums[j];
nums[j] = val;
if (i < j - 1) {
i++;
j--;
} else {
return i + 1;
}
}
return i;
}
}