LeetCode:1381. 设计一个支持增量操作的栈(中等)

题目描述

请你设计一个支持下述操作的栈。

实现自定义栈类 CustomStack：

• CustomStack(int maxSize)：用 maxSize 初始化对象，maxSize 是栈中最多能容纳的元素数量，栈在增长到 maxSize 之后则不支持 push 操作。
• void push(int x)：如果栈还未增长到 maxSize ，就将 x 添加到栈顶。
• int pop()：弹出栈顶元素，并返回栈顶的值，或栈为空时返回 -1 。
• void inc(int k, int val)：栈底的 k 个元素的值都增加val。如果栈中元素总数小于k ，则栈中的所有元素都增加 val。

示例：
输入：
[“CustomStack”,“push”,“push”,“pop”,“push”,“push”,“push”,“increment”,“increment”,“pop”,“pop”,“pop”,“pop”]
[[3],[1],[2],[],[2],[3],[4],[5,100],[2,100],[],[],[],[]]
输出：
[null,null,null,2,null,null,null,null,null,103,202,201,-1]
解释：
CustomStack customStack = new CustomStack(3); // 栈是空的 []
customStack.push(1); // 栈变为 [1]
customStack.push(2); // 栈变为 [1, 2]
customStack.pop(); // 返回 2 --> 返回栈顶值 2，栈变为 [1]
customStack.push(2); // 栈变为 [1, 2]
customStack.push(3); // 栈变为 [1, 2, 3]
customStack.push(4); // 栈仍然是 [1, 2, 3]，不能添加其他元素使栈大小变为 4
customStack.increment(5, 100); // 栈变为 [101, 102, 103]
customStack.increment(2, 100); // 栈变为 [201, 202, 103]
customStack.pop(); // 返回 103 --> 返回栈顶值 103，栈变为 [201, 202]
customStack.pop(); // 返回 202 --> 返回栈顶值 202，栈变为 [201]
customStack.pop(); // 返回 201 --> 返回栈顶值 201，栈变为 []
customStack.pop(); // 返回 -1 --> 栈为空，返回 -1

题解

使用数组模拟栈，用一个变量 top 来记录当前栈顶的位置。

• 对于 push 操作，首先判断当前元素的个数是否达到上限，如果没有达到，就把 top 后移一个位置并添加一个元素。
• 对于 pop 操作，首先判断当前栈是否为空，非空返回栈顶元素并将 top 前移一位，否则返回 -1。
• 对于increment操作，直接对栈底的最多 k 个元素加上 val。
  额外申请一个数组add, 记录每一次增加操作的位置(即从栈底元素到该元素均需要实现加val元素), 每一个元素的加法操作可根据increment操作数量进行叠加。

class CustomStack {
public:
    vector<int> stk,add;
    int top;
    CustomStack(int maxSize) {
        stk.resize(maxSize);
        add.resize(maxSize);
        top = -1;
    }

    void push(int x) {
        if (top != stk.size() - 1){
            ++top;
            stk[top] = x;
        }    
    }
    
    int pop() {
        if (top == -1) return -1;
        int ret = stk[top] + add[top];
        if(top != 0){
            add[top-1] += add[top];//叠加多次增加操作后每个元素需要增加的量
        }
        add[top] = 0;
        --top;
        return ret;
        
    }
    
    void increment(int k, int val) {
        int lim = min(k-1,top);
        if(lim >= 0){
            add[lim] += val;//记录每一次执行increment操作,需要执行增加操作的元素的最大索引
        }        
    }
};
/**
 * Your CustomStack object will be instantiated and called as such:
 * CustomStack* obj = new CustomStack(maxSize);
 * obj->push(x);
 * int param_2 = obj->pop();
 * obj->increment(k,val);
 */
 

复杂度分析

• 时间复杂度:O(1)。
• 空间复杂度:O(n), n为栈的最大长度。

class CustomStack {
    vector<int> stk;
    int top;
public:
    CustomStack(int maxSize) {
        stk.resize(maxSize);
        top = -1;
    }
    
    void push(int x) {
        if(top != stk.size()-1 ){
            stk[++top] = x;  
        }
    }
    
    int pop() {
        if( top == -1){
            return -1;
        }
        return stk[top--];
    }
    
    void increment(int k, int val) {
        for( int i = 0 ;i < k && i <= top; i++ ){
            stk[i] += val;
        }
    }
};

复杂度分析

• 时间复杂度:push 操作和 pop 操作的时间复杂度为 O(1)，increment 操作的时间复杂度为 O(k)。
• 空间复杂度:O(n), n为栈的最大长度。

参考

设计一个支持增量操作的栈
1381. Python3的思路及其优化方案