查看題目詳情可點擊此處。
題目
給定一個數組 nums 和一個值 val，你需要原地移除所有數值等于 val 的元>素，返回移除后數組的新長度。
不要使用額外的數組空間，你必須在修改輸入數組并在使用 O(1) 額外空間>的條件下完成。
元素的順序可以改變。你不需要考慮數組中超出新長度后面的元素。
示例 1:
給定 nums = [3,2,2,3], val = 3,
函數應該返回新的長度 2, 并且 nums 中的前兩個元素均為 2。
示例 2:
給定 nums = [0,1,2,2,3,0,4,2], val = 2,
函數應該返回新的長度 5, 并且 nums 中的前五個元素為 0, 1, 3, 0, 4。
解題思路
題目中未要求數組元素的輸出順序，所以可以直接遍歷數組，當數組元素與val相等時，就將此數組元素與數組末尾元素交換，同時數組的末尾下標前移，直到遍歷到數組末尾結束。代碼如下：
public static int removeElement(int[] nums, int val) {
int len = nums.length;
if(nums.length < 1) {return len;}
for(int i = 0; i < len; i++) {
if(nums[i] == val) {
swap(nums, i, --len);
i--;
}
}
return len;
}
public static void swap(int[] nums, int beforeIndex, int afterIndex) {
int temp = nums[afterIndex];
nums[afterIndex] = nums[beforeIndex];
nums[beforeIndex] = temp;
}
這樣的方式也可以解題，但是不是穩定的算法，數組中的元素因為與數組末尾元素對調后會導致元素前后順序的混亂。如果要將滿足條件，也就是不等于val的數組元素按照未操作前的前后順序輸出，需要調整思路。
可以使用兩個指針，一個指針用以遍歷數組，一個指針用以標記經操作后的數組的末尾長度，遍歷到數組中與val相等的元素，便將相鄰且不等于val的元素與標記的數組末尾下標元素交換，數組末尾下標后移，遍歷至數組末尾結束。代碼如下：
class Solution {
public int removeElement(int[] nums, int val) {
int lastIndex = 0;
int maxIndex = nums.length - 1;
for(int i = 0; i < maxIndex; i++) {
if(nums[i] == val && nums[i + 1] == val) {
continue;
} else if(nums[i] == val && nums[i + 1] != val) {
swap(nums, lastIndex, i + 1);
}
lastIndex++;
}
return lastIndex;
}
public void swap(int[] nums, int beforeIndex, int afterIndex) {
int temp = nums[afterIndex];
nums[afterIndex] = nums[beforeIndex];
nums[beforeIndex] = temp;
}
}
這段代碼解決方案沒問題，但在遇到邊界情況時會出現問題，出錯情況如下。
輸入：
[2]
3
輸出：
[]
預期：
[2]
當數組元素只有一個時會便會出現問題，原因在于數組遍歷只會遍歷到數組末尾的前一位，而此時返回的數組末尾長度仍然是0，導致輸出空數組，于是我打算使用哨兵的方式解決，將數組末尾值用val值替換出來，在遍歷完之后，再單獨對被替換出的數組末尾值判斷。
class Solution {
public int removeElement(int[] nums, int val) {
int lastIndex = 0;
if(nums.length < 1) {return lastIndex;}
int maxIndex = nums.length - 1;
int lastEle = nums[maxIndex];
nums[maxIndex] = val;
for(int i = 0; i < maxIndex; i++) {
if(nums[i] == val && nums[i + 1] == val) {
continue;
} else if(nums[i] == val && nums[i + 1] != val) {
swap(nums, lastIndex, i + 1);
}
lastIndex++;
}
if(lastEle != val) {
nums[lastIndex] = lastEle;
lastIndex++;
}
return lastIndex;
}
public void swap(int[] nums, int beforeIndex, int afterIndex) {
int temp = nums[afterIndex];
nums[afterIndex] = nums[beforeIndex];
nums[beforeIndex] = temp;
}
}
