删除元素（LintCode）

删除元素

给定一个数组和一个值，在原地删除与值相同的数字，返回新数组的长度。

元素的顺序可以改变，并且对新的数组不会有影响。

样例

给出一个数组 [0,4,4,0,0,2,4,4]，和值 4

返回 4 并且4个元素的新数组为[0,0,0,2]

思路就是把除删除元素外的元素移到最左边，r是最右端非被删除元素，可以在一开始就找到最右端非被删除元素。用i遍历数组，找到最左端被删除元素，把r所指元素赋到i位置即可。代码里我是先找最左端被删除元素，然后再找最右端非被删除元素。时间复杂度是相同的。

 public class Solution {
     /**
      *@param A: A list of integers
      *@param elem: An integer
      *@return: The new length after remove
      */
     public int removeElement(int[] A, int elem) {
         int r = A.length - 1;
         if(r < 0) return 0;
         int i = 0;

         while(r > i) {
             if(A[i] == elem) {
                 while(A[r] == elem && r > 0) r--;
                 if(r > i) {
                     A[i] = A[r];
                     r--;
                 }
             }
             i++;
         }

         if(A[r] == elem) r--;

         return r + 1;
     }
 }