劍指offer之面試題23:從上往下打印二叉樹

題目:從上往下打印出二叉樹的每一個節點,同層節點從左至右打印。(BFS)ide

思路:spa

根據題目,想到層序遍歷時,按照從上到下,從左到右的訪問每個結點,用到一個輔助隊列(先進先出才能保證從左到右的訪問)。隊列

代碼:
it

/*
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) 
    {
	//保存打印序列
        vector<int> v1;
        //先進先出
        queue<TreeNode *> q1;
        if(root==NULL)
        {
            return v1;
        }
        q1.push(root);
        v1.push_back(root->val);
        while(!q1.empty())
        {
            TreeNode* tmp=q1.front();
            q1.pop();
            if(tmp->left!=NULL)
            {
                q1.push(tmp->left);
                v1.push_back(tmp->left->val);
            }
            if(tmp->right!=NULL)
            {
                q1.push(tmp->right);
                v1.push_back(tmp->right->val);
            }
        }
        return v1;
    }
};
相關文章
相關標籤/搜索