Computer >> Máy Tính >  >> Lập trình >> C ++

N-ary Tree Traversal đặt hàng trước trong C ++

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ư

N-ary Tree Traversal đặt hàng trước trong C ++

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