Leetcode: Remove Element

Remove Element | LeetCode OJ

We need to remove the given element and shift the rest of the elements.

1. Initialize index at 0
2. Fill all the elements except the input number while increasing the index

Remarks:

1. O(1) space complexity
2. O(n) time complexity

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        j = 0
        for n in nums:
            if n != val:
                nums[j] = n
                j += 1
        return j