Scribbling

LeetCode: 27. Remove Element 본문

Computer Science/Coding Test

LeetCode: 27. Remove Element

focalpoint 2022. 1. 17. 11:36
class Solution:
    def removeElement(self, nums: List[int], val: int) -> int: 
        length = len(nums)
        i = 0
        while i <= length - 1:
            if nums[i] == val:
                nums[i], nums[length-1] = nums[length-1], nums[i]
                length -= 1
            else:
                i += 1
        return length