两个栈实现队列的pop和push

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

解题思路

两个栈实现队列：栈2为空——将栈1元素组个弹出压入栈2，栈2顶层元素就是最先进入栈1的元素；栈2不为空——栈2栈顶元素就是最先进入队列的元素。那么删除操作就是pop栈2中元素；添加操作就很简单——栈1push元素。

代码

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int pLen=pre.size();
        if(pLen==0){
            return NULL;
        }
        vector <int> pre_left,pre_right,vin_left,vin_right;
        //define a queal position gen
        int gen=0;
        for(int i=0;i<pLen;++i){
            if(pre[0]==vin[i]){
                gen=i;
                break;
            }
        }
        TreeNode* root=new TreeNode(pre[0]);
        for(int i=0;i<gen;++i){
            pre_left.push_back(pre[i+1]);
            vin_left.push_back(vin[i]);
        }
        for(int i=gen+1;i<pLen;++i){
            pre_right.push_back(pre[i]);
            vin_right.push_back(vin[i]);
        }
        root->left=reConstructBinaryTree(pre_left, vin_left);
        root->right=reConstructBinaryTree(pre_right, vin_right);
        return root;
    }
};