题目:给你一个长度为length的数组arr和一个值val,请原地移除所有数值等于val的元素,并返回移除后数组的新长度。
解题思路:快慢指针,通过将后方的元素覆盖当前值为val的元素来实现。
int removeElement(int* arr, int len, int val){ //数组arr,数组长度len,目标值val
int k = 0;
// 两个指针i和k,i遍历原数组,k维护结果数组
for(int i = 0; i < len; i ++)
if(arr[i] != val) arr[k ++] = arr[i];
return k; //返回结果数组的长度
}
全部代码如下:
#include<iostream>
#include<cstdio>
using namespace std;
int removeElement(int* arr, int len, int val){ //数组arr,数组长度len,目标值val
int k = 0;
// 两个指针i和k,i遍历原数组,k维护结果数组
for(int i = 0; i < len; i ++)
if(arr[i] != val) arr[k ++] = arr[i];
return k; //返回结果数组的长度
}
int main(){
int arr[] = {1, 2, 2, 2, 3, 4, 5, 6, 7, 8};
printf("原数组:\n");
for(int i = 0; i < 10; i ++) printf("%d ", arr[i]);
int k = removeElement(arr, 10, 2);
printf("\n");
printf("结果数组:\n");
for(int i = 0; i < k; i ++) printf("%d ", arr[i]);
}
更多代码请参考:手撕考研数据结构(代码汇总)