Day_29.py

class CustomStack:

    def __init__(self, maxSize: int):
        self.stack = []
        self.maxSize = maxSize

    def push(self, x: int) -> None:
        if len(self.stack) >= self.maxSize:
            return 
        self.stack.append(x)

    def pop(self) -> int:
        if len(self.stack) <= 0 :
            return -1
        lastElement = self.stack[-1]
        self.stack.pop()
        return lastElement 
        
    def increment(self, k: int, val: int) -> None:
        if len(self.stack) < k:
            for i in range (len(self.stack)):
                self.stack[i]+=val
        else :
            for i in range(k):
                self.stack[i]+=val


# Your CustomStack object will be instantiated and called as such:
# obj = CustomStack(maxSize)
# obj.push(x)
# param_2 = obj.pop()
# obj.increment(k,val)