【LeetCode】Remove Element

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int result=0;
        for(int i=0;i<n;i++)
        {
            A[result]=A[i];
            if(A[i]!=elem)
            {
                result++;
            }
        }
        return result;
    }
};