025 Remove Element.py

"""
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.
Author : Rajeev Ranjan
"""

class Solution:
    def removeElement_negative_index(self, A, elem):
        """
        Constant space
        Algorithms: Two Pointers
        Partitioning the array into 3 parts, closed, open, back
        Data structure: array
        :param A: list
        :param elem: integer
        :return: "shrunk" list
        """

        open_ptr = 0
        back_ptr = -1  # Python style backward
        while len(A)+back_ptr>=open_ptr:
            if A[open_ptr]==elem:
                A[open_ptr], A[back_ptr] = A[back_ptr], A[open_ptr]
                back_ptr -= 1
            else:
                open_ptr += 1

        return len(A)+back_ptr+1  # length is index+1

    def removeElement(self, A, elem):
        """
        Constant space
        Algorithms: Two Pointers
        Partitioning the array into 3 parts, closed, open, back
        Data structure: array
        :param A: list
        :param elem: integer
        :return: "shrunk" list
        """
        open_ptr = 0
        end_ptr = len(A)
        while open_ptr<end_ptr:
            if A[open_ptr]==elem:
                end_ptr -= 1
                A[open_ptr], A[end_ptr] = A[end_ptr], A[open_ptr]
            else:
                open_ptr += 1

        return end_ptr


if __name__=="__main__":
    A = [1, 3, 4, 2, 5, 4]
    elem = 4
    solution = Solution()
    assert solution.removeElement(A, elem)==solution.removeElement_negative_index(A, elem)