[Leetcode][Python]27: Remove Element

最新推荐文章于 2026-08-23 17:45:32 发布

转载最新推荐文章于 2026-08-23 17:45:32 发布 · 115 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

原文链接：http://www.cnblogs.com/Dabay/p/4261424.html

收录于

当前文章被以下社区和专栏收录：

本文介绍了解决LeetCode上的第27题“移除元素”的算法思路。该题要求从数组中移除指定元素的所有实例，并返回新数组的长度。通过使用双指针技巧，一个指针用于记录插入位置，另一个用于遍历数组，有效地实现了元素的移除。文章详细解释了算法流程，并提供了Python实现代码。

# -*- coding: utf8 -*-
'''
__author__ = 'dabay.wang@gmail.com'

27: Remove Element
https://oj.leetcode.com/problems/remove-element/

Given an array and a value, remove all instances of that value in place and return the new length.
The order of elements can be changed. It doesn't matter what you leave beyond the new length.

===Comments by Dabay===
一次循环，两个指针，一个指向插入的位置，另外一个一直往前面走。
    如果二号指针的数就是需要删除的数，二号指针继续走。
    如果不是要删除的数，把二号指针指向的数移到一号指针的位置上，然后两个指针继续走。
最后跟新数组，返回长度。
'''

class Solution:
    # @param    A       a list of integers
    # @param    elem    an integer, value need to be removed
    # @return an integer
    def removeElement(self, A, elem):
        i = j = 0
        while j < len(A):
            if A[j] != elem:
                A[i] = A[j]
                i += 1
            j += 1
        A = A[:i]
        return len(A)


def main():
    sol = Solution()
    print sol.removeElement([1,1,2], 1)


if __name__ == "__main__":
    import time
    start = time.clock()
    main()
    print "%s sec" % (time.clock() - start)