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, ]
廣告