JZ32 從上往下列印二叉樹

蓝色的海嗷發表於2024-04-19

image

image

/*
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) {

        //層序遍歷 列印所有的值
        //佇列實現 先將root放入佇列,判斷佇列是否為空 不為空列印
        //再判斷是否有孩子,一次將孩子從左到右入佇列列印 後將root出戰
        queue<TreeNode*> q;
        vector<int> v;
        if (root != nullptr) {
            q.push(root);
        }
        while (q.empty() == false) { //q不為空
            v.push_back(q.front()->val);
            if (q.front()->left != nullptr) {
                q.push(q.front()->left);
            }
            if (q.front()->right != nullptr) {
                q.push(q.front()->right);
            }
            //刪除最先入佇列的節點
            q.pop();
        }
        return v;
    }
};

相關文章