剑指Offer - 九度1368 - 二叉树中和为某一值的路径
2013-11-23 03:46
- 题目描述:
-
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
- 输入:
-
每个测试案例包括n+1行:
第一行为2个整数n,k(1<=n<=10000),n表示结点的个数,k表示要求的路径和,结点编号从1到n。
接下来有n行。这n行中每行为3个整数vi,leftnode,rightnode,vi表示第i个结点的值,leftnode表示第i个结点的左孩子结点编号,rightnode表示第i个结点的右孩子结点编号,若无结点值为-1。编号为1的结点为根结点。
- 输出:
-
对应每个测试案例,先输出“result:”占一行,接下来按字典顺序输出满足条件的所有路径,这些路径由结点编号组成,输出格式参照输出样例。
- 样例输入:
-
5 22 10 2 3 5 4 5 12 -1 -1 4 -1 -1 7 -1 -1 1 5 1 -1 -1
- 样例输出:
-
result: A path is found: 1 2 5 A path is found: 1 3 result:
题意分析:
题目要求找出从根节点到叶节点的路径之和为指定值的所有路径。思路很清楚:DFS遍历,vector存储路径,sum记录路径和。
所有节点遍历一次,时间复杂度O(n)。遍历过程中需要存储路径,vector的空间复杂度为O(log(n))。
1 // 652977 zhuli19901106 1368 Accepted 点击此处查看所有case的执行结果 1112KB 1552B 30MS 2 // 201311180013 3 #include <cstdio> 4 #include <vector> 5 using namespace std; 6 7 void dfs(const int a[][3], const int &n, int r, int sum, int target, vector<int> &path) 8 { 9 if(r < 1 || r > n){ 10 return; 11 } 12 13 path.push_back(r); 14 15 if(a[r][1] == -1 && a[r][2] == -1){ 16 if(sum + a[r][0] == target){ 17 // a valid path is found. 18 printf("A path is found:"); 19 for(int i = 0; i < path.size(); ++i){ 20 printf(" %d", path[i]); 21 } 22 printf("\n"); 23 } 24 } 25 26 if(a[r][1] != -1){ 27 dfs(a, n, a[r][1], sum + a[r][0], target, path); 28 } 29 if(a[r][2] != -1){ 30 dfs(a, n, a[r][2], sum + a[r][0], target, path); 31 } 32 33 path.pop_back(); 34 } 35 36 int main() 37 { 38 const int MAXN = 10005; 39 int n, i, k, r; 40 int a[MAXN][3]; 41 int c[MAXN]; 42 vector<int> path; 43 44 while(scanf("%d%d", &n, &k) == 2){ 45 for(i = 1; i <= n; ++i){ 46 c[i] = 0; 47 } 48 for(i = 1; i <= n; ++i){ 49 scanf("%d%d%d", &a[i][0], &a[i][1], &a[i][2]); 50 if(a[i][1] > 0){ 51 ++c[a[i][1]]; 52 } 53 if(a[i][2] > 0){ 54 ++c[a[i][2]]; 55 } 56 if(a[i][1] > 0 && a[i][2] > 0 && a[i][1] > a[i][2]){ 57 // do this swap to make sure the result is in lexicological order 58 a[i][1] ^= a[i][2] ^= a[i][1] ^= a[i][2]; 59 } 60 } 61 r = -1; 62 for(i = 1; i <= n; ++i){ 63 if(c[i] == 0){ 64 r = i; 65 break; 66 } 67 } 68 printf("result:\n"); 69 path.clear(); 70 dfs(a, n, r, 0, k, path); 71 path.clear(); 72 } 73 74 return 0; 75 }
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步