Leetcode 589.N叉树的前序遍历(简单)递归、非递归实现

题目描述:

给定一个 N 叉树,返回其节点值的前序遍历。

例如,给定一个 3叉树 :

 

 

返回其前序遍历: [1,3,5,6,2,4]。

 

说明: 递归法很简单,你可以使用迭代法完成此题吗?

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal

思路:1.非递归,用单独的栈来记录子树的情况,类比二叉树的非递归遍历可以实现。注意先右子树入栈再左子树。

           2.递归

/*
// 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> result;
        stack<Node*> stk; //stack的类型是指向Node的指针,开始写stack<Node>错
        if(root == NULL) return result;
        stk.push(root);
        while(!stk.empty())
        {
            Node* tmp= stk.top();
            result.push_back(tmp->val);
            stk.pop();
            for(int i=tmp->children.size()-1; i>=0; --i)
            {
                stk.push(tmp->children[i]);
            }
        }
        return result;
    }
};
/*
// 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> result;
    vector<int> preorder(Node* root) {    
        if(root != NULL){
            result.push_back(root->val);//先序根节点,然后递归遍历每一个子树
            for(auto child : root->children){
                preorder(child);
            }
        }
        return result;
    }
};
/*
// 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:
    void preord(Node *root, vector<int>& t)
    {
        if(root == NULL)  return;
        t.push_back(root->val);
        for(auto p: root->children)
            preord(p,t);
    }
    vector<int> preorder(Node* root) {
        vector<int> result;    
        preord(root,result);
        return result;
    }
};

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值