leetcode第27题-Remove Element

#include<stdio.h>
#include<stdlib.h>

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

int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		int *a=(int *)malloc(sizeof(int)*n);
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		int elem;
		scanf("%d",&elem);
		int newLen=removeElement(a,n,elem);
		printf("%d\n",newLen);
	}
	return 0;
}