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

题目来源：计蒜客

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

亲爱的小伙伴们，题目是不是很简单呢？
提示： int removeElement(int A[], int n, int elem)
其中，n代表数组长度，elem代表要删掉的元素。


格式：
输入一个数n，继而输入一个数组A[n],接着输入要删除的元素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;
}