LeetCode之Remove Element

int removeElement(int A[], int n, int elem) //Remove Element{int num=0;for (int i=0;i<n-num;i++){if (A[i]==elem){while(i!=n-num-1){if(A[n-num-1]!=elem) {A[i]=A[n-num-1];A[n-num-1]=NULL;break;}else {A[n-num-1]=NULL;num++;}}if(i==n-num-1) A[i]=NULL;num++;}}return n-num;}