二叉树路径求和的问题可以有多种变种,但最常见的是找到从根节点到叶子节点的路径,使得路径上所有节点的值之和等于给定目标和。我们可以使用深度优先搜索(DFS)的策略来解决这个问题。以下是求解二叉树路径求和问题的基本思路:
从根节点开始,使用深度优先搜索遍历二叉树。
当遍历到某个节点时,将该节点的值从目标和中减去。
如果当前节点是叶子节点(即左右子节点都为空),检查剩余的目标和是否为 0。如果为 0,说明找到了一条满足条件的路径。
对于非叶子节点,递归地对其左右子节点进行相同的操作。
遍历完成后,返回所有满足条件的路径。
《算法笔记》原题:
https://sunnywhy.com/sfbj/9/2/340
#include <iostream>
#include <vector>
using namespace std;
const int MAXN = 51;
struct Node{
int l,r;
int data;
}nodes[MAXN];
int sum = 0;
void GetTreePathSum(int root, int nodeSum){
if(root == -1)return;
nodeSum += nodes[root].data;
if(nodes[root].l == -1 && nodes[root].r == -1){
sum += nodeSum;
} else{
GetTreePathSum(nodes[root].l, nodeSum);
GetTreePathSum(nodes[root].r, nodeSum);
}
}
int main(){
int n,L,R,data;
cin >> n;
for (int i = 0; i < n; ++i) {
cin >> data;
nodes[i].data = data;
}
for (int i = 0; i < n; ++i) {
cin >> L >> R;
nodes[i].l = L;
nodes[i].r = R;
}
GetTreePathSum(0,0);
cout << sum;
}