LeetCode面试题 03.03 堆盘子

于 2025-05-16 07:30:00 发布
阅读量178 收藏
点赞数 8
分类专栏： LeetCode 文章标签： leetcode java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jackie_xiaonan/article/details/147907507
版权

题目

解答

package leetcode.editor.cn;

import java.util.LinkedList;
import java.util.List;

//leetcode submit region begin(Prohibit modification and deletion)
class StackOfPlates {
    private List<List<Integer>> stackSet = new LinkedList<>();
    private int capacity = 0;

    public StackOfPlates(int cap) {
        this.capacity = cap;
    }

    public void push(int val) {
        if (this.capacity <= 0) {
            return;
        }

        List<Integer> stack = null;
        if (stackSet.isEmpty()) {
            stack = new LinkedList<>();
            stackSet.addLast(stack);
        } else {
            stack = stackSet.getLast();
            if (stack.size() >= capacity) {
                stack = new LinkedList<>();
                stackSet.addLast(stack);
            }
        }
        stack.addLast(val);
    }

    public int pop() {
        if (stackSet.isEmpty()) {
            return -1;
        }

        List<Integer> stack = null;
        while (!stackSet.isEmpty()) {
            stack = stackSet.getLast();
            if (!stack.isEmpty()) {
                int value = stack.removeLast();
                if (stack.isEmpty()) {
                    stackSet.removeLast();
                }
                return value;
            }
            stackSet.removeLast();
        }

        return -1;
    }

    public int popAt(int index) {
        if (stackSet.size() < index) {
            return -1;
        }

        List<Integer> stack = null;

        while (!stackSet.isEmpty() && stackSet.size() > index) {
            stack = stackSet.get(index);
            if (!stack.isEmpty()) {
                int value = stack.removeLast();
                if (stack.isEmpty()) {
                    stackSet.remove(index);
                }
                return value;
            }
            stackSet.remove(index);
        }

        return -1;
    }
}

/**
 * Your StackOfPlates object will be instantiated and called as such:
 * StackOfPlates obj = new StackOfPlates(cap);
 * obj.push(val);
 * int param_2 = obj.pop();
 * int param_3 = obj.popAt(index);
 */
//leetcode submit region end(Prohibit modification and deletion)

测试用例

package leetcode.editor.cn;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class StackOfPlatesTest {
    private StackOfPlates s = null;
    
    @Test
    public void test1() {
        s = new StackOfPlates(1);
        s.push(1);
        s.push(2);

        int value = 0;
        value = s.popAt(1);
        Assert.assertTrue(value == 2);

        value = s.pop();
        Assert.assertTrue(value == 1);

        value = s.pop();
        Assert.assertTrue(value == -1);
    }

    @Test
    public void test2() {
        s = new StackOfPlates(2);
        s.push(1);
        s.push(2);
        s.push(3);

        int value = 0;
        value = s.popAt(0);
        Assert.assertTrue(value == 2);

        value = s.popAt(0);
        Assert.assertTrue(value == 1);

        value = s.popAt(0);
        Assert.assertTrue(String.valueOf(value), value == 3);
    }

    @Test
    public void test3() {
        s = new StackOfPlates(2);
        s.push(1);
        s.push(2);
        s.push(3);

        int value = 0;
        value = s.popAt(1);
        Assert.assertTrue(value == 3);

        value = s.popAt(1);
        Assert.assertTrue(String.valueOf(value), value == -1);
    }

    @Test
    public void test4() {
        s = new StackOfPlates(2);
        s.push(1);
        s.push(2);
        s.push(3);

        int value = 0;
        value = s.pop();
        Assert.assertTrue(value == 3);

        value = s.pop();
        Assert.assertTrue(value == 2);

        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 1);
    }

    /**
     * input:["StackOfPlates", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "push", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop", "pop"]
     * [[0], [2], [8], [56], [1], [39], [40], [44], [63], [11], [38], [20], [55], [25], [14], [11], [1], [20], [16], [6], [18], [3], [39], [45], [2], [22], [64], [6], [30], [39], [3], [19], [63], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], [], []]
     * Output:[null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,63,19,3,39,30,6,64,22,2,45,39,3,18,6,16,20,1,11,14,25,55,20,38,11,63,44,40,39,1,56,8,2]
     * Expected:[null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,null,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1]
     * stdout:
     */
    @Test
    public void test5() {
        s = new StackOfPlates(0);
        s.push(2);
        s.push(8);
        s.push(56);
        s.push(1);
        s.push(39);
        s.push(40);

        int value = 0;
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);

        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);

        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);
    }

    /**
     * input:["StackOfPlates", "pop", "popAt", "push", "popAt", "popAt", "pop", "pop", "push", "popAt", "pop", "push", "push", "pop", "popAt", "popAt", "push", "push", "push", "popAt", "pop", "pop", "pop", "popAt", "pop", "push", "popAt", "push", "push", "popAt", "push", "push", "pop", "popAt", "push", "pop", "popAt", "push", "pop", "push", "pop", "popAt", "popAt", "pop", "push", "push", "pop", "popAt", "push", "push", "pop", "pop", "popAt"]
     * [[3], [], [1], [1], [2], [2], [], [], [9], [3], [], [51], [20], [], [2], [0], [35], [1], [19], [3], [], [], [], [1], [], [36], [1], [19], [3], [3], [15], [44], [], [3], [46], [], [0], [42], [], [31], [], [0], [2], [], [10], [49], [], [1], [14], [50], [], [], [3]]
     * Output:[null,-1,-1,null,-1,-1,1,-1,null,-1,9,null,null,20,-1,51,null,null,null,-1,19,1,35,-1,-1,null,-1,null,null,-1,null,null,44,-1,null,46,3,null,42,null,31,19,-1,15,null,null,49,10,null,null,50,14,-1]
     * Expected:[null,-1,-1,null,-1,-1,1,-1,null,-1,9,null,null,20,-1,51,null,null,null,-1,19,1,35,-1,-1,null,-1,null,null,-1,null,null,44,-1,null,46,3,null,42,null,31,19,-1,15,null,null,49,-1,null,null,50,14,-1]
     */

    @Test
    public void test6() {
        int value = 0;

//	"StackOfPlates"	[3]	null
        s = new StackOfPlates(3);
//	"pop"	[]	-1
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"popAt"	[1]	-1
        value = s.popAt(1);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[1]	null
        s.push(1);
//	"popAt"	[2]	-1
        value = s.popAt(2);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"popAt"	[2]	-1
        value = s.popAt(2);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"pop"	[]	1
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 1);
//	"pop"	[]	-1
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[9]	null
        s.push(9);
//	"popAt"	[3]	-1
        value = s.popAt(3);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"pop"	[]	9
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 9);
//	"push"	[51]	null
        s.push(51);
//	"push"	[20]	null
        s.push(20);
//	"pop"	[]	20
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 20);
//	"popAt"	[2]	-1
        value = s.popAt(2);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"popAt"	[0]	51
        value = s.popAt(0);
        Assert.assertTrue(String.valueOf(value), value == 51);
//	"push"	[35]	null
        s.push(35);
//	"push"	[1]	null
        s.push(1);
//	"push"	[19]	null
        s.push(19);
//	"popAt"	[3]	-1
        value = s.popAt(3);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"pop"	[]	19
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 19);
//	"pop"	[]	1
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 1);
//	"pop"	[]	35
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 35);
//	"popAt"	[1]	-1
        value = s.popAt(1);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"pop"	[]	-1
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[36]	null
        s.push(36);
//	"popAt"	[1]	-1
        value = s.popAt(1);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[19]	null
        s.push(19);
//	"push"	[3]	null
        s.push(3);
//	"popAt"	[3]	-1
        value = s.popAt(3);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[15]	null
        s.push(15);
//	"push"	[44]	null
        s.push(44);
//	"pop"	[]	44
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 44);
//	"popAt"	[3]	-1
        value = s.popAt(3);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[46]	null
        s.push(46);
//	"pop"	[]	46
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 46);
//	"popAt"	[0]	3
        value = s.popAt(0);
        Assert.assertTrue(String.valueOf(value), value == 3);
//	"push"	[42]	null
        s.push(42);
//	"pop"	[]	42
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 42);
//	"push"	[31]	null
        s.push(31);
//	"pop"	[]	31
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 31);
//	"popAt"	[0]	19
        value = s.popAt(0);
        Assert.assertTrue(String.valueOf(value), value == 19);
//	"popAt"	[2]	-1
        value = s.popAt(2);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"pop"	[]	15
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 15);
//	"push"	[10]	null
        s.push(10);
//	"push"	[49]	null
        s.push(49);
//	"pop"	[]	49
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 49);
//	"popAt"	[1]	-1
        value = s.popAt(1);
        Assert.assertTrue(String.valueOf(value), value == -1);
//	"push"	[14]	null
        s.push(14);
//	"push"	[50]	null
        s.push(50);
//	"pop"	[]	50
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 50);
//	"pop"	[]	14
        value = s.pop();
        Assert.assertTrue(String.valueOf(value), value == 14);
//	"popAt"	[3]	-1

        value = s.popAt(3);
        Assert.assertTrue(String.valueOf(value), value == -1);
    }

}

注意点

  • 栈的容量，有可能是异常值。
  • 当某个栈为空时，需要及时清理掉。
  • 当栈中没有元素或不存在该栈时，pop和popAt 应返回-1。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小南家的青蛙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值