用 cnt 統計 val 出現的次數,index是 val 之外的數組元素的新下標
class Solution {
public int removeElement(int[] nums, int val) {
int index = 0, cnt = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == val) {
cnt++;
} else {
nums[index++] = nums[i];
}
}
return nums.length - cnt;
}
}