题目:http://ac.jobdu.com/problem.php?cid=1039&pid=10
题目描述:
-
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
-
输入:
-
每个测试案例包括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:
本题一看挺简单的,题意就是在n个结点形成的二叉树中,找出从根结点到叶子结点的结点值之和等于所给出值k的路径数!做本题 的时候,由于太大意了,有很多地方没有注意,特别是题目要求路径要按字典序输出,写过过了样例后,一直交,就是WA,最后再仔细看题目才发现没按字典序输出!!以后做题要特别仔细了!
以下是AC代码:
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <cctype>
#include <cstdlib>
#include <string.h>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 10100
struct Node
{
int index;
int value;
int left;
int right;
};
struct Node p[N];
int path[N];
void dfs(int n,int k,int index,int sum,int c)
{
int i ;
if(sum > k)
return;
if((sum == k )&& (p[index].left == -1) && (p[index].right == -1))
{
printf("A path is found:");
for(i = 0;i <= c;i++)
printf(" %d",path[i]);
printf("\n");
return;
}
if(p[index].left != -1)
{
c++;
path[c] = p[p[index].left].index;
dfs(n,k,p[index].left,sum + p[p[index].left].value,c);
c--;
}
if(p[index].right != -1)
{
c++;
path[c] = p[p[index].right].index ;
dfs(n,k,p[index].right,sum + p[p[index].right].value,c);
c--;
}
return;
}
int main()
{
int n,k,i,sum;
int v , l , r;
while(scanf("%d %d",&n,&k) == 2)
{
memset(p,-1,sizeof(p));
for(i = 0;i < n;i++)
{
scanf("%d %d %d",&v,&l,&r);
p[i].index = i + 1;
p[i].value = v;
if( l < r)//一开始就是这里没注意,一直wa,这是按字典序输出的关键!
{
p[i].left = l;
p[i].right = r;
}
else
{
p[i].left = r;
p[i].right = l;
}
if(p[i].left != -1)
p[i].left--;
if(p[i].right != -1)
p[i].right--;
}
printf("result:\n");
sum = p[0].value;
path[0] = 1;
dfs(n,k,0,sum,0);
}
return 0;
}