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;
}
发布了36 篇原创文章 · 获赞 0 · 访问量 372
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 数字20 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览