Leetcode 1381. Design a stack that supports incremental operations

Title

1381. Design a stack supporting incremental operation

Topic link

https://leetcode-cn.com/problems/design-a-stack-with-increment-operation/

Topic ideas

Python3 Simulating stack operations with arrays ( The first in, the second out )

code for Python3

class CustomStack:

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

    def push(self, x: int) -> None:
        if len(self.arr) < maxSize:
            self.arr.append(x)

    def pop(self) -> int:
        if self.arr:
            return self.arr.pop()
        else:
            return -1


    def increment(self, k: int, val: int) -> None:
        if not self.arr:
            return
        elif len(self.arr) <= k:
            for i in range(len(self.arr)):
                self.arr[i] += val
        else:
            for i in range(k):
                self.arr[i] += val

Copy

Complexity analysis

• Time complexity : O(N)
• Spatial complexity : O(1)