FilesExpand file tree

master

/

remove-element.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

17 lines (15 loc) · 444 Bytes

master

/

remove-element.py

Copy path

Top

File metadata and controls

• Code
• Blame

17 lines (15 loc) · 444 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

# Time: O(n)

# Space: O(1)

class Solution(object):

# @param A a list of integers

# @param elem an integer, value need to be removed

# @return an integer

def removeElement(self, A, elem):

i, last = 0, len(A) - 1

while i <= last:

if A[i] == elem:

A[i], A[last] = A[last], A[i]

last -= 1

else:

i += 1

return last + 1