移除元素,大致看一下描述你会觉得很简单,不是和顺序表操作是一个道理嘛。但是,题目里的==“原地”和“O(1)”== 是不是格外显眼?仔细一想就会发现并不是有手就行。
注意复杂度要求:空间复杂度为O(1),我们只能以常数的空间来保存若干变量。如果像我们顺序表一样搞删插操作,我们虽然不需要考虑数组中超出新长度后面的元素(事例是允许的)。在 realloc,malloc时意味着开辟额外空间,势必会打破空间复杂度为1的原则,所以我们要另辟蹊径。
我们既要做到遍历数组,又要达到定位删除的工作,不难想到一种手段——双指针:
int removeElement(int* nums, int numsSize, int val){ int* p = nums; int* q = nums; int size = numsSize; for(;p<numsSize;p++) { if(*p==val) { size--; //对删除对象进行计数 } else { *q = *p; *q++;//非删除对象依次保留 } return size; }
我们也可以搞不需要指针的双指针,区间 [0,q) 中的元素都不等于 val。当左右指针遍历完输入数组以后,q 的值就是输出数组的长度。
int removeElement(int* nums, int numsSize, int val){ int p = 0; int q = 0; for(q=0;q<numsSize;q++) { if(q!=val) { nums[p] =nums[q]; p++; } } return p; }
但是,在val元素较少时,我们可以尝试不整体挪,我们可以以换位的方式进行覆盖更替,比如我想删除数组中 {0,1,2,3,4,5} 中的元素 0,我们可以用最右边的 5进行替代成{5,1,2,3,4}这样效率就会大大提升,如下:
int removeElement(int* nums, int numsSize, int val){ int p = 0; int q = numsSize; while(p<q) { if(nums[p]==val) { nums[p]=nums[q-1]; q--; } else p++; } return numsSize; }