ในปัญหานี้ เราได้รับ Binary tree และ a sum S. และเราต้องหาเส้นทางที่เริ่มต้นจาก root ไปยังโหนดใดๆ ของ tree ซึ่งให้ผลรวมเท่ากับผลรวมที่กำหนด
ป้อนข้อมูล
Sum = 14 Output : path : 4 10 4 3 7
เพื่อหาวิธีแก้ปัญหานี้ เราจำเป็นต้องค้นหาการข้ามผ่านของการสั่งซื้อล่วงหน้าของไบนารีทรี แล้วหาเส้นทางที่รวมกันเป็นยอดที่กำหนด
ตัวอย่าง
#include<bits/stdc++.h>
using namespace std;
struct Node{
int key;
struct Node *left, *right;
};
Node* insertNode(int key){
Node* temp = new Node;
temp->key = key;
temp->left = temp->right = NULL;
return (temp);
}
void printPathsUtilSum(Node* curr_node, int sum, int
sum_so_far, vector<int> &path){
if (curr_node == NULL)
return;
sum_so_far += curr_node->key;
path.push_back(curr_node->key);
if (sum_so_far == sum ){
for (int i=0; i<path.size(); i++)
cout<<path[i]<<"\t";
cout<<endl;
}
if (curr_node->left != NULL)
printPathsUtilSum(curr_node->left, sum,
sum_so_far, path);
if (curr_node->right != NULL)
printPathsUtilSum(curr_node->right, sum,
sum_so_far, path);
path.pop_back();
}
void pathWithSum(Node *root, int sum){
vector<int> path;
printPathsUtilSum(root, sum, 0, path);
}
int main (){
Node *root = insertNode(4);
root->left = insertNode(10);
root->right = insertNode(3);
root->right->left = insertNode(7);
root->right->right = insertNode(1);
root->left->left = insertNode(8);
root->left->right = insertNode(6);
int sum = 14;
cout<<"Paths with the given sum are : "<<endl;
pathWithSum(root, sum);
return 0;
} ผลลัพธ์
เส้นทางที่มีผลรวมที่กำหนดคือ −
4 10 4 3 7
