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;
}