LeetCode面试题 03.01: 三合一

原创于 2020-05-12 21:36:21 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 同时被 2 个专栏收录

63 篇文章

订阅专栏

栈

7 篇文章

订阅专栏

本文详细介绍了如何在一个固定大小的数组中实现三个独立的栈，通过巧妙地分配内存地址，确保每个栈都能在其特定的空间内进行压栈和弹栈操作，同时提供了完整的代码示例，展示了该数据结构的设计思路和具体实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

First. Problem’s Description

在这里插入图片描述

Second. Problem’s Solution

I don’t think there is any point that’s difficult to solve.

Three. Code For Solution

class TripleInOne {
private:
    int *Data;
    int sSize = -1;
    unsigned int s_begin[3];
    unsigned int s_cur[3] = {0, 0, 0};   //how many elem in stack
public:
    TripleInOne(int stackSize) {
        auto tmp = new int[stackSize * 3]{};
        if(!tmp)    exit(0);
        Data = tmp;
        sSize = stackSize;
        s_begin[0] = 0;
        s_begin[1] = stackSize;
        s_begin[2] = 2 * stackSize;
    }
    
    void push(int stackNum, int value) {
        if(s_cur[stackNum] < sSize)    Data[s_begin[stackNum] + s_cur[stackNum]++] = value;
    }
    
    int pop(int stackNum) {
        return !s_cur[stackNum] ? -1 : Data[s_begin[stackNum] + --s_cur[stackNum]];
    }
    
    int peek(int stackNum) {
        return !s_cur[stackNum] ? -1 : Data[s_begin[stackNum] + s_cur[stackNum] - 1];
    }
    
    bool isEmpty(int stackNum) {
        return s_cur[stackNum] == 0;
    }
};