LeetCode Remove Element删除元素

 1 class Solution {
 2 public:
 3 int removeElement(int A[], int n, int elem) {
 4     int *p=A,*e=&A[n-1];
 5     int i,num=n;
 6     for(i=0;i<n;i++){    //一共要对比n次，不能用n来处理，会影响循环
 7         if(*p==elem){
 8             if(p==e)    //已经处理到最后一个相同，只需总数减1
 9                 num--;
10             else{
11                 *p=*e;
12                 e--;
13                 num--;
14             }
15         }
16         else
17             p++;
18     }
19     return num;
20 }
21 };