题目:
https://leetcode-cn.com/problems/remove-element/
题解一:
暴力循环求解,一层for循环遍历数组,一层for循环做更新数组。时间复杂度:O(n^2)
public int removeElement(int[] nums, int val) {
int size = nums.length;
for (int i = 0; i < size; i++) {
// 发现需要移除的元素,就把后面元素依次向前移动一位
if (nums[i] == val) {
for (int j = i; j < size - 1; j++) {
nums[j] = nums[j+1];
}
// 此时数组的长度-1
size--;
// 因为num[i]的数组被覆盖了,所以num[i]的值需要从新判断,所以这里i需要 -1
i--;
}
}
return size;
}
题解二:
双指针法:快指针用来遍历数组,慢指针用来重置数组,只要 nums[fastIndex] != val 就把当前元素保存,否则跳过;重复这一过程直到数组结尾。时间复杂度:O(n)
public int removeElement(int[] nums, int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.length; fastIndex++) {
if (nums[fastIndex] != val) {
nums[slowIndex++] = nums[fastIndex];
}
}
return slowIndex;
}
题解三:
虽然双指针算法能在O(n)的复杂度内解决此问题,但是对于要删除的数据很少时,会做过多的复制操作,例如:num=[1,2,3,5,4],val=4;会对前4个元素做不必要的复制操作。另一例子:num=[4,1,2,3,5],val=4;也没有必要将后面4个元素都向左移动一位,因为题目描述可以更改元素顺序,所以我们只需要将4移动到数组末尾,释放最后一个元素让数组长度减一即可。
public int removeElement(int[] nums, int val) {
int length = nums.length;
int index = 0;
while (index < length) {
if (nums[index] == val) {
nums[index] = nums[length - 1];
length--;
} else {
index++;
}
}
return length;
}
时间复杂度:O(n)。
在这个算法中要复制的次数等于要删除元素的个数,所以当要删除元素个数较少时效率会高一点。