题目1368：二叉树中和为某一值的路径

题目描述：

输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

输入：

每个测试案例包括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 2210 2 35 4 512 -1 -14 -1 -17 -1 -11 51 -1 -1

样例输出：

result:A path is found: 1 2 5A path is found: 1 3result:

答疑：

解题遇到问题?分享解题心得?讨论本题请访问： http://t.jobdu.com/thread-8091-1-1.html

 

/*思路：想法很简单，细节难把握，首先建立二叉树，然后调用fun()函数进行排序(
因为输出要字典排序),如果左孩子的mark,大于右孩子的mark，交换左右孩子，
最后进行深搜，用数组模拟栈。*/
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#define MAX 10004 
using namespace std ;
typedef struct node
{
	int data , mark;
	struct node *left , *right ;
}node , *tree ;

void Init(tree A[])
{
	for( int i = 0 ; i < MAX ; i ++){
		A[i] = (node *)malloc(sizeof(node));
		A[i]->left = A[i]->right = NULL ;
		A[i]->mark = -1 ;
	}
	return ;
}
int save[MAX] , k ;
int num ;
void print()
{
	int i= 0 ;
	printf("A path is found: ");
	for( i = 0 ; i < k-1 ;i ++)
		printf("%d ",save[i] +  1);
	printf("%d\n",save[k-1]+1);
}
void Find(tree T , int sum )
{
	if(T){
		save[k++] = T->mark ;
		sum +=T->data ;
		
		Find(T->left ,sum);
		Find(T->right,sum );
		if(sum == num && T->left == NULL && T->right == NULL ) //细节必须是叶子节点
			print();
		k-- ;
		sum -=T->data ;
	}
}
void fun(tree &T){
	if(T)
	{
		if((T->left && T->right) && T->left->mark >T->right->mark) 
		{
			tree p = T->left ;
			T->left = T->right ;
			T->right = p ;
		}
		fun(T->left);
		fun(T->right);
	}
}
int main(void)
{
	int i ,n ;
	while(scanf("%d %d",&n,&num)!=EOF){
		tree A[MAX] ;
		Init(A);
		k = 0 ;
		for( i = 0 ; i < n ; i++)
		{
			int data , l , r ;
			scanf("%d %d %d",&data , &l , &r);
			A[i]->data = data ;
			A[i]->mark = i ;
			A[i]->left = A[i]->right = NULL ;
			if(l!=-1)
				A[i]->left = A[l-1];
			if(r!=-1)
				A[i]->right = A[r-1] ;
		}
		printf("result:\n");
		int sum = 0 ;
		fun(A[0]);
		Find(A[0] , sum);
	}
	return 0 ;
}