LeetCode 589. N 叉树的前序遍历 & 590. N 叉树的后序遍历 & 429. N 叉树的层序遍历

题目:

【589:https://leetcode.cn/problems/n-ary-tree-preorder-traversal/description/

【590:https://leetcode.cn/problems/n-ary-tree-postorder-traversal/description/

【429:https://leetcode.cn/problems/n-ary-tree-level-order-traversal/

        给定一个 n 叉树的根节点 root ,返回 其节点值的前序遍历或后序遍历或层序遍历

示例:

输入:root = [1,null,3,2,4,null,5,6]

前序遍历输出:[1,3,5,6,2,4]
后序遍历输出:[5,6,3,2,4,1]
层序遍历输出:[[1],[3,2,4],[5,6]]

代码:

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    // 前序遍历
    vector<int> preorder(Node* root) {
        vector<int> res;
        traverseForPreOrder(root, res);
        return res;  
    }
    void traverseForPreOrder(Node* node, vector<int>& res) {
        if (node == nullptr) {
            return;
        }

        res.emplace_back(node->val);
        for (auto ch : node->children) {
            traverseForPreOrder(ch, res);
        }
    }


    // 后序遍历
    vector<int> postorder(Node* root) {
        vector<int> res;
        traverseForPostOrder(root, res);
        return res;
    }
    void traverseForPostOrder(Node* node, vector<int>& res) {
        if (node == nullptr) {
            return;
        }

        for (auto child : node->children) {
            traverseForPostOrder(child, res);
        }
        res.push_back(node->val);
    }

    // 层序遍历
    vector<vector<int>> levelOrder(Node* root) {
        if (root == nullptr) {
            return {};
        }

        vector<vector<int>> res;
        queue<Node*> q;
        q.push(root);

        while (!q.empty()) {
            vector<int> curLevel;
            int curLevelSize = q.size();

            for (int i = 0; i < curLevelSize; i++) {
                auto curNode = q.front();
                q.pop();
                curLevel.push_back(curNode->val);
                for (auto child : curNode->children) {
                    q.push(child);
                }
            }
            res.push_back(curLevel);
        }
        return res;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

m0_38045911

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值