LeetCode Remove Element

int removeElement(int A[], int n, int elem) {int i=0,j=0;while(j<n&&A[j]!=elem){i++;j++;}if (j==n)return n;while(j<n){while(j<n&&A[j]==elem) j++;if (j<n){A[i++]=A[j++];}}return i;}