【计蒜客系列】挑战难题9：元素移除

题目来源：计蒜客

如：A[] = {1, 2, 3, 4, 5}, 要删除数字3, 那么返回数组长度为4.

亲爱的小伙伴们，题目是不是很简单呢？

提示： int removeElement(int A[], int n, int elem)

其中，n代表数组长度，elem代表要删掉的元素。

格式：

输入一个数n，继而输入一个数组A
,接着输入要删除的元素elem，返回剩余数组长度index.

样例1

输入：

2

3 3

3

输出：
0

#include<stdio.h>
#include<malloc.h>

int removeElement(int A[], int n, int elem)
{
int i;
int len = n;
for(i=0;i<n;i++)
{
if(A[i] == elem)
len--;
}
return len;
}

int main(int argc, char **argv) {
int i,n,elem;
scanf("%d",&n);
int *a = (int *)malloc(sizeof(int)*n);
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&elem);
int ans = removeElement(a,n,elem);
printf("%d\n",ans);
return 0;
}