LintCode 172---删除元素

最新推荐文章于 2019-05-26 22:46:47 发布

转载最新推荐文章于 2019-05-26 22:46:47 发布 · 128 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/cnmoti/p/10828511.html

博客给出了 LintCode 172 题删除元素的代码实现。代码定义了一个 Solution 类，其中的 removeElement 方法接收整数数组 A 和要删除的整数 elem，通过双指针法交换元素位置，最终返回删除元素后的新数组长度。

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) {
          // write your code here
        if(A.length == 0)
            return 0;
            
        int font = 0;
        int end = A.length-1;
            
        while(font <= end){
                        // Move value to position
            if(A[font] == elem && A[end] != elem){
                // swap
                A[font++] = A[end--];
            }else{
                if(A[font] != elem)
                    font++; 
                if(A[end] == elem)
                    end--;
            }
        }
        return font;
    }
}