标签：int top pop 1381 public 增量 push 设计 stack

用栈实现

import java.util.Stack;

public class Algorithm {

    public static void main(String[] args) {

        CustomStack myStack = new CustomStack(5);

        for (int i = 0; i < 5; i++) {
            myStack.push(i);
        }

        myStack.increment(2, 10);

        for (int i = 0; i < 5; i++) {
            System.out.println(myStack.pop());
        }
    }
}

class CustomStack {

    Stack<Integer> stack;
    int maxSize;

    public CustomStack(int maxSize) {

        stack = new Stack();
        this.maxSize = maxSize;
    }

    public void push(int x) {

        if (stack.size() < maxSize) {

            stack.push(x);
        }
    }

    public int pop() {

        return stack.isEmpty() ? -1 : stack.pop();
    }

    public void increment(int k, int val) {

        if (!stack.isEmpty()) {

            Stack<Integer> temStack = new Stack<>();

            /**
             * 使用辅助栈，将元素全部转移，再按k个数进行加值
             */
            while (stack.size() > 0) {
                temStack.push(stack.pop());
            }

            if (temStack.size() <= k) {

                while (temStack.size() > 0) {
                    stack.push(temStack.pop() + val);
                }
            }
            else {
                while (temStack.size() > 0) {

                    if (stack.size() < k) {
                        stack.push(temStack.pop() + val);
                    } else {
                        stack.push(temStack.pop());
                    }
                }
            }
        }
    }
}

用数组实现

public class Algorithm {

    public static void main(String[] args) {

        CustomStack myStack = new CustomStack(5);

        for (int i = 0; i < 5; i++) {
            myStack.push(i);
        }

        myStack.increment(2, 10);

        for (int i = 0; i < 5; i++) {
            System.out.println(myStack.pop());
        }
    }
}

class CustomStack {

    int[] array;
    int tem[];
    int top;

    public CustomStack(int maxSize) {

        /**
         * top为栈顶位置加一
         */
        top = 0;

        /**
         * 使用两个数组实现，元素值改变时先不动元素
         * 用辅助数组存放要改变元素的最后一个位置，和要改变的值
         * 当取走一个被改变的元素后，将位置减一
         */
        array = new int[maxSize];
        tem = new int[maxSize + 1];
    }

    public void push(int x) {

        if (top < array.length) {

            array[top] = x;
            top++;
        }
    }

    public int pop() {

        if (top == 0){
            return -1;
        }

        int res = array[top - 1];

        /**
         * 辅助数组多一个空间
         * 当辅助数组对应的栈顶位置不为零时，说明这个元素增值了，该位置的元素就是增量
         * 加上以后，再将增量赋值给前一个位置，也就是等到pop时再增，否则不动
         */
        if (tem[top] != 0) {

            res += tem[top];

            /**
             * 注意这个+=
             * 因为可能有多次增量，所以值要累加
             */
            tem[top - 1] += tem[top];
            tem[top] = 0;
        }

        top--;
        return res;
    }

    public void increment(int k, int val) {

        /**
         * 有增量时，只记录最后一个变化元素的索引和增量
         * 其前面的元素暂时不增加
         */
        tem[Math.min(k, top)] += val;
    }
}

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

标签：int,top,pop,1381,public,增量,push,设计,stack
来源： https://www.cnblogs.com/taoyuann/p/15410664.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。