Giả sử chúng ta có một cây n-ary, chúng ta phải tìm đường đi ngang của các nút của nó.
Vì vậy, nếu đầu vào giống như
thì đầu ra sẽ là [1,3,5,6,2,4]
Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -
-
Xác định ans mảng
-
Xác định một phương thức được gọi là preorder (), phương thức này sẽ có giá trị root
-
nếu gốc là null, thì -
-
trả về danh sách trống
-
-
chèn giá trị gốc vào cuối ans
-
cho tất cả i con trong mảng con của root
-
đặt hàng trước (i)
-
-
trả lại ans
Ví dụ
Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -
#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));
} Đầu vào
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); Đầu ra
[1, 3, 5, 6, 2, 4, ]