【計蒜客系列】挑戰難題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;
}