class Solution {
public:
int removeElement(int A[], int n, int elem) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
if (A==NULL||n<=0){
return 0;
}
int index=0;
int i=0;
while(i<n){
if (A[i]==elem) {
i++;
} else {
A[index++]=A[i++];
}
}
return index;
}
};
public:
int removeElement(int A[], int n, int elem) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
if (A==NULL||n<=0){
return 0;
}
int index=0;
int i=0;
while(i<n){
if (A[i]==elem) {
i++;
} else {
A[index++]=A[i++];
}
}
return index;
}
};