1. 程式人生 > >【計蒜客系列】挑戰難題9:元素移除

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