Leetcode 1381. 设计一个支持增量操作的栈

题目名称

1381. 设计一个支持增量操作的栈

题目链接

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

题目思路

Python3用数组模拟栈操作(先进后出的特点)

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

复制

复杂度分析

• 时间复杂度: O(N)
• 空间复杂度: O(1)