【数据结构·考研】移除数组中值等于val的元素

题目:给你一个长度为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]);
}

 更多代码请参考:手撕考研数据结构(代码汇总)

猜你喜欢

转载自blog.csdn.net/cjw838982809/article/details/130991411