码迷,mamicode.com
首页 > 其他好文 > 详细

N-ary Tree Postorder Traversal

时间:2020-03-16 13:07:10      阅读:45      评论:0      收藏:0      [点我收藏+]

标签:color   flag   init   order   return   cto   efi   while   ini   

迭代

/*
// 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> postorder(Node* root) {
        vector<int> vec;
        stack<Node*> s;
        unordered_map<Node*,bool> m;
        Node* node=root;
        while(node||!s.empty()) {
            while(node) {
                s.push(node);
                m[node]=true;
                bool flag=false;
                for(auto& child:node->children) {
                    if(!m.count(child)) {node=child;flag=true;break;}
                }
                if(!flag) node=nullptr;
            }
            if(!s.empty()) {
                node=s.top();
                bool flag=false;
                for(auto& child:node->children) {
                    if(!m.count(child)) {node=child;flag=true;break;}
                }
                if(!flag) {
                    node=nullptr;
                    vec.push_back(s.top()->val);
                    s.pop();
                }
            }
        }
        return vec;
    }
};

 递归

class Solution {
public:
    vector<int> postorder(Node* root) {
        vector<int> vec;
        postorder_helper(root,vec);
        return vec;
    }
    void postorder_helper(Node* node,vector<int>& vec) {
        if(!node) return;
        for(auto& child:node->children) {
            postorder_helper(child,vec);
        }
        vec.push_back(node->val);
    }
};

 

N-ary Tree Postorder Traversal

标签:color   flag   init   order   return   cto   efi   while   ini   

原文地址:https://www.cnblogs.com/LiuQiujie/p/12502873.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!