二叉树中和为某一值的路径

题目

题目描述：
输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
输入：
每个测试案例包括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:

思路

简单的树的前序遍历的变形，可以用栈记录每个节点的大小，够了unique_sum打印即可

AC代码

#include <stdio.h>
#include <stdlib.h>
 
struct btree
{
    int value;
    int loc, lchild, rchild;
};
 
struct stack
{
    int top;
    struct btree data[10005];
};
 
struct stack *s;
 
void print_uniquesum_path(struct btree *bt, int root, int unique)
{
    int lchild, rchild, i, sum;
 
    if (root == -1) {
        return;
    }
 
    s->data[s->top ++] = bt[root];
 
    lchild = bt[root].lchild;
    rchild = bt[root].rchild;
 
 
    if (lchild == -1 && rchild == -1) {
        for (i = sum = 0; i < s->top; i ++) {
            sum += s->data[i].value;
        }
        if (sum == unique) {
            printf("A path is found:");
            for (i = 0; i < s->top; i ++) {
                printf(" %d", s->data[i].loc);
            }
            printf("\n");
        }
    }
 
    if (lchild < rchild) {
        print_uniquesum_path(bt, lchild, unique);
        print_uniquesum_path(bt, rchild, unique);
    } else {
        print_uniquesum_path(bt, rchild, unique);
        print_uniquesum_path(bt, lchild, unique);
    }
 
    s->top --;
}
 
int main()
{
    int i, n, k;
    struct btree *bt;
 
    while (scanf("%d %d", &n, &k) != EOF) {
        // 初始化树
        bt = (struct btree *)malloc(sizeof(struct btree) * (n + 1));
        for (i = 1; i <= n; i ++) {
            scanf("%d %d %d", &bt[i].value, &bt[i].lchild, &bt[i].rchild);
            bt[i].loc = i;
        }
 
        // 初始化栈
        s = (struct stack *)malloc(sizeof(struct stack));
        s->top = 0;
 
        // 遍历树
        printf("result:\n");
        print_uniquesum_path(bt, 1, k);
        free(bt);
    }
 
    return 0;
}
/**************************************************************
    Problem: 1368
    User: wangzhengyi
    Language: C
    Result: Accepted
    Time:30 ms
    Memory:4112 kb
****************************************************************/

 

转载于:https://www.cnblogs.com/snake-hand/archive/2013/06/06/3122996.html