力扣题号:27.移除元素
一、题目描述
原地移除数组中所有等于val的元素,要求不能使用额外的辅助空间,即空间复杂度为O(1)。返回移除后新数组的size。
二、示例
数组为[4,5,6,4,4,4]。
移除的元素为4。
原地移除元素4之后的数组为[5,6],数组长度为2。
三、求解思路
实现数组元素删除有两种方法,第一种是暴力解法,使用两个for循环;第二种方法是使用双指针法(快慢指针法)。
四、代码实现
#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
// 暴力解法
int violentSolution(int* arr, int length, int val) {
for (int i = 0; i < length; i++) {
if (arr[i] == val) {
for (int j = i + 1; j < length; j++) {
arr[j - 1] = arr[j];
}
i--; // 重新判断i位置元素是否等于val
length--; // 数组的长度-1
}
}
return length;
}
// 双指针法(快慢指针法)
int doublePointer(int* arr, int length, int val) {
int slow = 0;
for (int fast = 0; fast < length; fast++) {
if (arr[fast] != val) {
arr[slow++] = arr[fast];
}
}
return slow;
}
// 打印新的数组
void print(int *arr, int length) {
for (int i = 0; i < length; i++) {
printf("%d ", arr[i]);
}
printf("\n");
}
int main() {
int arr[10] = {1,3,4,5,6,7,5,4,3,4};
int val = 3;
int size = violentSolution(arr, 10, val);
printf("The size is %d\n", size);
print(arr, size);
printf("\n");
int arr1[10] = { 1,3,4,5,6,7,5,4,3,4 };
val = 5;
size = doublePointer(arr1, 10, val);
printf("The size is %d\n", size);
print(arr1, size);
return 0;
}