題目9：用兩個棧實現佇列

用兩個棧來實現一個佇列，完成佇列的Push和Pop操作。 佇列中的元素為int型別。 

難度⭐

思路

stack1 正向佇列 用於新增。

stack2 反向佇列 用於刪除。

我的程式碼。

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

    int pop() { 
        if(stack2.size()<=0)
        {
            while(stack1.size()>0)
            {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        if(stack2.size()==0)
        {
            throw exception();
        }
        int a=stack2.top();
        stack2.pop();
        return a;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};

vs2005上編譯。

注意 T& 的含義，代表引用。

模板的用法。

#pragma once
#include <stack>
#include <exception>

using namespace std;

template <typename T> class CQueue
{
public:
    CQueue(void);
    ~CQueue(void);
    
    // 在佇列末尾新增一個結點
    void appendTail(const T& node);

    // 刪除佇列的頭結點
    T deleteHead();

private:
    stack<T> stack1;
    stack<T> stack2;
};

template <typename T> CQueue<T>::CQueue(void)
{
}

template <typename T> CQueue<T>::~CQueue(void)
{
}

 //我的程式碼
template<typename T> void CQueue<T>::appendTail(const T& element)
{
	 stack1.push(element);
} 

template<typename T> T CQueue<T>::deleteHead()
{
	if(stack2.size()<=0)
	{
		//先從頭開始壓入棧中
		while(stack1.size()>0)
		{
			stack2.push(stack1.top());
			stack1.pop();
		}
	}
	if (stack2.size()==0)
	{
		throw exception(); 
	}
	
	T& head=stack2.top();
	stack2.pop();
 

	return head ;
}

 

python

增#直接push 到 stack1

刪#如果stack2 是空的就從stack1 取數（while)，然後pop stack2.否則異常

class Solution:
    def __init__(self):
        self.stackA = []
        self.stackB = []
    def push(self, node):
        self.stackA.append(node)
    def pop(self):
        if self.stackB:
            return self.stackB.pop()
        elif self.stackA == []:
            return None
        else:
            while self.stackA:
                self.stackB.append(self.stackA.pop())
            return self.stackB.pop()

 

java

刪#對stack2判空，如果空就取，不空就.pop(.top())

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
        if(stack1.empty()&&stack2.empty())
        {
            throw new RuntimeException("Queue is empty!");
        }
        if(stack2.empty())
        {
            while(!stack1.empty())
            {
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

相關題目

兩個佇列描述一個棧