C语言--移除元素

移除元素

给定一个数组 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;
}