C++ 中 N 元树先序遍历

假设我们有一棵 N 元树，则必须找到其节点的前序遍历顺序。

因此，如果输入如下

则输出将是 [1,3,5,6,2,4]

为解决此问题，我们将按照下列步骤进行 -

  • 定义一个数组 ans

  • 定义一个名为 preorder() 的方法，这将获取 root

  • 如果 root 为空，则 -

    • 返回空列表

  • 在 ans 的末尾插入 root 的值

  • 对于 root 的 children 数组中的所有子节点 i

    • preorder(i)

  • 返回 ans

示例 

让我们看看下面的实现来加深理解 -

 现场演示

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
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&g; ans;
   vector<int> preorder(Node* root) {
      if (!root)
         return {};
      ans.emplace_back(root->val);
      for (auto i : root->children)
         preorder(i);
      return ans;
   }
};
main(){
   Solution ob;
   Node *node5 = new Node(5), *node6 = new Node(6);
   vector<Node*> child_of_3 = {node5, node6};
   Node* node3 = new Node(3, child_of_3);
   Node *node2 = new Node(2), *node4 = new Node(4);l
   vector<Node*> child_of_1 = {node3, node2, node4};
   Node *node1 = new Node(1, child_of_1);
   print_vector(ob.preorder(node1));
}

输入

Node *node5 = new Node(5), *node6 = new Node(6);
vector<Node*> child_of_3 = {node5, node6};
Node* node3 = new Node(3, child_of_3);
Node *node2 = new Node(2), *node4 = new Node(4);
vector<Node*> child_of_1 = {node3, node2, node4};
Node *node1 = new Node(1, child_of_1);

输出

[1, 3, 5, 6, 2, 4, ]

更新于： 2020 年 6 月 11 日

525 次浏览

开启你的 职业生涯

通过完成该课程获得认证

开始
广告