Trong hướng dẫn này, chúng ta sẽ viết một chương trình tìm số lượng nút lá cho mọi nút trong cây n-ary.
Cho một cây n-ary, chúng ta phải tìm số nút lá cho mọi cây con. Hãy xem một ví dụ.
Đầu vào
N = 8 tree = [[2, 3], [], [4, 5, 6], [7, 8], [], [], [], []]
Đầu ra
1->5 2->1 3->4 4->2 5->1 6->1 7->1 8->1
Thuật toán
-
Khởi tạo n-ary tree với cây bạn thích.
-
Sử dụng DFS để đi qua cây.
-
Duy trì một mảng để lưu trữ số lượng mỗi nút lá nút.
-
Tăng số lượng nút lá sau cuộc gọi đệ quy tới DFS.
-
In tất cả các nút với số lượng nút lá.
Thực hiện
Sau đây là cách thực hiện thuật toán trên trong C ++
#include <bits/stdc++.h> using namespace std; void insertNode(int x, int y, vector<int> tree[]) { tree[x].push_back(y); } void DFS(int node, int leaf[], int visited[], vector<int> tree[]) { leaf[node] = 0; visited[node] = 1; for (auto it : tree[node]) { if (!visited[it]) { DFS(it, leaf, visited, tree); leaf[node] += leaf[it]; } } if (!tree[node].size()) { leaf[node] = 1; } } int main() { int N = 8; vector<int> tree[N + 1]; insertNode(1, 2, tree); insertNode(1, 3, tree); insertNode(3, 4, tree); insertNode(3, 5, tree); insertNode(3, 6, tree); insertNode(4, 7, tree); insertNode(4, 8, tree); int leaf[N + 1]; int visited[N + 1]; for (int i = 0; i <= N; i++) { visited[i] = 0; } DFS(1, leaf, visited, tree); for (int i = 1; i <= N; i++) { cout << i << "->" << leaf[i] << endl; } return 0; }
Đầu ra
Nếu bạn chạy đoạn mã trên, thì bạn sẽ nhận được kết quả sau.
1->5 2->1 3->4 4->2 5->1 6->1 7->1 8->1