1. 題目

三合一。描述如何只用一個數組來實現三個棧。

你應該實現push(stackNum, value)、pop(stackNum)、isEmpty(stackNum)、peek(stackNum)方法。stackNum 表示棧下標，value 表示壓入的值。

構造函數會傳入一個stackSize參數，代表每個棧的大小。

示例1:輸入： ["TripleInOne", "push", "push", "pop", "pop", "pop", "isEmpty"] [[1], [0, 1], [0, 2], [0], [0], [0], [0]]輸出： [null, null, null, 1, -1, -1, true] 說明：當棧為空時`pop, peek`返回-1，當棧滿時`push`不壓入元素。示例2:輸入： ["TripleInOne", "push", "push", "push", "pop", "pop", "pop", "peek"] [[2], [0, 1], [0, 2], [0, 3], [0], [0], [0], [0]]輸出： [null, null, null, null, 2, 1, -1, -1]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/three-in-one-lcci
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

class TripleInOne {int n, v;int size;vector<int> tail;//尾指針位置vector<int> stk; public:TripleInOne(int stackSize) {n = stackSize;size = 3*stackSize;stk.resize(size);tail.resize(3);tail[0] = 0, tail[1] = stackSize, tail[2] = 2*stackSize;}void push(int stackNum, int value) {if(tail[stackNum] < (stackNum+1)*n ){stk[tail[stackNum]] = value;tail[stackNum]++;}}int pop(int stackNum) {if(tail[stackNum] == stackNum*n)return -1;v = stk[tail[stackNum]-1];tail[stackNum]--;return v;}int peek(int stackNum) {if(tail[stackNum] == stackNum*n)return -1;return stk[tail[stackNum]-1];}bool isEmpty(int stackNum) {return tail[stackNum]==stackNum*n;} };

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 03.01. 三合一（数组栈）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。