移除元素
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
不使用额外的数组空间。
例如:给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
#include<stdio.h>
int removeElement(int* nums, int numsSize, int val) {
int i, j = 0;
int tmp = 0;
j = numsSize;
for (i = 0; i < numsSize; i++) {//记录移除后元素的个数
if (nums[i] == val) { //用j记因为numsSzie下边还要用
j--;
}
}
int n = 0; //用来指向要移动元素的后一个元素
for (i = 0; i < numsSize; i++) {
if (nums[i] == val) {
tmp = i;//找到第一个值等于val的元素
n = tmp + 1;
if (n >= numsSize) {
break;//如果tmp是最后一个元素,就结束循环,不移动了
}
while (1) {
if (n >= numsSize) {
break;//这个是防止那个值等于val的元素移出数组,造成堆栈溢出
}
if (nums[tmp] != nums[n]) {//交换
int m = nums[tmp];
nums[tmp] = nums[n];
nums[n] = m;
break;
}
else if (n < numsSize && nums[tmp] == nums[n]) {
n++;//如果连着两个或者多个值等于val的元素,那n就往后拨
}
}
}
}
numsSize = j;
return numsSize;
}
int main() {
int nums[3] = {0,1,2,2,3,0,4,2};
int val = 2;
int numsSize = sizeof(nums)/sizeof(nums[0]);
int a = removeElement(nums, numsSize, val);
for (int i = 0; i < a; i++) {
printf("%d ", nums[i]);
}
return 0;
}