美文网首页
lintcode 172. Remove Element

lintcode 172. Remove Element

作者: cuizixin | 来源:发表于2018-08-27 18:32 被阅读33次

难度:

1. Description

172. Remove Element

2. Solution

  • python
    用list的remove函数
class Solution:
    """
    @param: A: A list of integers
    @param: elem: An integer
    @return: The new length after remove
    """
    def removeElement(self, A, elem):
        # write your code here
        for i in range(A.count(elem)):
            A.remove(elem)
        return len(A)

不用list的remove函数

class Solution:
    """
    @param: A: A list of integers
    @param: elem: An integer
    @return: The new length after remove
    """
    def removeElement(self, A, elem):
        # write your code here
        j = len(A)-1
        for i in range(len(A)-1, -1, -1):
            if A[i] == elem:
                A[j], A[i] = A[i], A[j]
                j -= 1 
        return j+1 

3. Reference

  1. https://www.lintcode.com/problem/remove-element/description

相关文章

网友评论

      本文标题:lintcode 172. Remove Element

      本文链接:https://www.haomeiwen.com/subject/rkvtwftx.html