ในบทช่วยสอนนี้ เราจะเขียนโปรแกรมที่ค้นหาจำนวนโหนดปลายสุดสำหรับทุกโหนดในแผนผัง n-ary
จากต้นไม้ n-ary เราต้องหาจำนวน leaf nodes สำหรับทุก subtree มาดูตัวอย่างกัน
ป้อนข้อมูล
N = 8 tree = [[2, 3], [], [4, 5, 6], [7, 8], [], [], [], []]
ผลผลิต
1->5 2->1 3->4 4->2 5->1 6->1 7->1 8->1
อัลกอริทึม
-
เริ่มต้นต้นไม้ n-ary ด้วยต้นไม้ที่คุณชอบ
-
ใช้ DFS เพื่อสำรวจผ่านต้นไม้
-
รักษาอาร์เรย์เพื่อเก็บจำนวนโหนดโหนดปลายสุดแต่ละโหนด
-
เพิ่มจำนวนโหนดปลายสุดหลังจากการเรียกซ้ำไปยัง DFS
-
พิมพ์โหนดทั้งหมดที่มีโหนดลีฟนับ
การนำไปใช้
ต่อไปนี้เป็นการนำอัลกอริธึมข้างต้นไปใช้ใน 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; }
ผลลัพธ์
หากคุณเรียกใช้โค้ดด้านบน คุณจะได้ผลลัพธ์ดังต่อไปนี้
1->5 2->1 3->4 4->2 5->1 6->1 7->1 8->1