LeetCode N叉树的前序遍历(递归、递推)

给定一个 N 叉树,返回其节点值的前序遍历。
例如,给定一个 3叉树 :
在这里插入图片描述
返回其前序遍历: [1,3,5,6,2,4]。
说明: 递归法很简单,你可以使用迭代法完成此题吗?
思路分析: 和二叉树的前序遍历思想一致,先访问根节点,在访问子节点。
请先翻阅 LeetCode 二叉树的前序遍历(递归、递推)

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

    Node() {}

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
public:
    vector<int> result;
    vector<int> preorder(Node* root) {
        myPreorder(root);
        return result;
    }
    //先序遍历:先根节点,再子节点
    void myPreorder(Node* root){
        if (root == NULL){
            return;
        }
        result.push_back(root->val);//先根节点
        for (auto child : root->children){//再子节点
            myPreorder(child);
        }
    }
};

在这里插入图片描述
方法二:递推法。

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

    Node() {}

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
public:
	vector<int> preorder(Node* root) {
		vector<int> result;
		stack<Node*> myStack;//辅助栈
		while (root != NULL || !myStack.empty()) {
			if (root == NULL) {
				root = myStack.top();
				myStack.pop();
			}
			//先序遍历:先根节点,再子节点
			result.push_back(root->val);//先根节点
            //注意栈是先进后出,所以需要倒置放入所有子节点
			for (auto it = root->children.rbegin(); it != root->children.rend(); ++it) {
				myStack.push(*it);
			}
			root = NULL;
		}
		return result;
	}
};

在这里插入图片描述

相关推荐
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页