剑指offer：栈的压入、弹出序列 && 剑指offer：从上到下从左到右打印二叉树

栈的压入、弹出序列题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。
假设压入栈的所有数字均不相等。

例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

（注意：这两个序列的长度是相等的）

思路

• 定义一个栈来模拟栈的压入和弹出过程
  • 如果压栈的数等于出栈的数，则可以直接弹出
  • 如果出栈的数等于栈顶元素，则弹出栈顶元素

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        if(pushV.size() != popV.size() || pushV.size() == 0) return false;
        stack<int> st;
        int j = 0;
        for(int i = 0; i < pushV.size();i++){
            if(popV[j] == pushV[i]){
                j++;
                continue;
            }
            while(!st.empty() &&  st.top() == popV[j]){
                st.pop();
                j++;
            }
            st.push(pushV[i]);
        }
        
        while(!st.empty() && popV[j] == st.top()){
            st.pop();
            j++;
        }
        return st.empty();
    }
};

从上到下从左到右打印二叉树：题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

思路

• 定义一个队列
• 打印当层节点的值，并将当层节点的左子树和右子树加入队列

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        if(root == nullptr) return {};
        queue<TreeNode*> que;
        
        vector<int> res;
        que.push(root);
        while(!que.empty()){
            int size = que.size();
            while(size--){
                res.push_back(que.front()->val);
                if(que.front()->left) que.push(que.front()->left);
                if(que.front()->right) que.push(que.front()->right);
                que.pop();
            }
        }
        return res;

    }
};