面试题25 二叉树中和为某一值的路径

地址：http://ac.jobdu.com/problem.php?pid=1368

题目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

解题思路：

1、前序遍历二叉树

2、每遍历一个节点，在数组中记录其序号，将其值加入sum变量中

3、当节点既没有左孩子又没有右孩子时，该节点为叶子节点，判断sum值是否为k，为k则打印数组中的节点序号

4、遍历完某节点（包括其左右子树）后，从数组中删除该节点序号。sum变量也减去该节点的值

注意：

题目要求按字典序输出序列。例如：

输入：
5 22
10 3 2
12 -1 -1
5 4 5
7 -1 -1
7 -1 -1

输出：
result:
A path is found: 1 2
A path is found: 1 3 4
A path is found: 1 3 5

解决方案：

当 left > right 时，交换这两个节点。

<span style="font-size:14px;">#include <stdio.h>
#include <stdlib.h>
 
int sum;
int buf[10000][3];
int s[10000];
int top;
 
struct treeNode {
    int data;
    int num;
    treeNode *left;
    treeNode *right;
};
 
void print(){
    printf("A path is found:");
    for(int i = 0; i <= top; i ++) {
        printf(" %d", s[i]);
    }
    printf("\n");
}
 
void traverse(treeNode *root, int k) {
    if(root == NULL)
        return;
    sum += root->data;
    s[++ top] = root->num;
    if(root->left == NULL && root->right == NULL){
        if(sum == k)
            print();
        s[-- top];
        sum -= root->data;
        return;
    }
    if(root->left != NULL)
        traverse(root->left, k);
    if(root->right != NULL)
        traverse(root->right, k);
    s[-- top];
    sum -= root->data;
    return;
}
 
//创建二叉树
//参数i：节点元素的序号，从0开始
treeNode* createTree(int i) {
    treeNode *root = (treeNode *)malloc(sizeof(treeNode));
    root->data = buf[i][0];
    root->num = i + 1;
    if (buf[i][1] != -1)
        root->left = createTree(buf[i][1] - 1);
    else
        root->left = NULL;
    if (buf[i][2] != -1)
        root->right = createTree(buf[i][2] - 1);
    else
        root->right = NULL;
    return root;
}
int main(){
    int n, k;
    while(scanf("%d %d", &n, &k) != EOF) {
        int left, right;
        for(int i = 0; i < n; i ++) {
            scanf("%d %d %d", &buf[i][0], &left, &right);
            if(left > right) {
                int tmp = left;
                left = right;
                right = tmp;
            }
            buf[i][1] = left;
            buf[i][2] = right;
        }
        treeNode *root = createTree(0);
        printf("result:\n");
        sum = 0;
        top = -1;
        traverse(root, k);
    }
    return 0;
}
/**************************************************************
    Problem: 1368
    Language: C++
    Result: Accepted
    Time:40 ms
    Memory:1704 kb
****************************************************************/</span>