给一棵点带权(权值各不相同,都是小于10000的正整数)的二叉树的中序序列和后序遍历,找一个叶子使得它到根的路径上的权和最小。如果有多解,该叶子本身的权值应尽量小。
输入
每两行表示一棵树,其中第一行为中序遍历,第二行为后序遍历。
输出
输出叶子的权值
样例输入
3 2 1 4 5 7 6
3 1 2 5 6 7 4
7 8 11 3 5 16 12 18
8 3 11 7 16 18 12 5
255
255
样例输出
1
3
255
解题思路
这个题明显就是考查二叉树的递归遍历,只要掌握了dfs的主要技巧,本题其实不难。主要需要解决的是利用题目给的中序后序序列构造成一棵二叉树,然后利用递归遍历求出权值进行保存以及判断。
因输入没有个数 所以利用字符进行处理,代码内有单独函数
以下是代码,以c为例:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
int s1[100], s2[100];
typedef struct Node TreeNode;
struct Node {
TreeNode *Left;
TreeNode *Right;
int data;
};
TreeNode* build_Tree(int L1, int R1, int L2, int R2) //构建二叉树
{
if (L1 > R1 || L2 > R2)
return NULL;
int i;
TreeNode *p = NULL;
p = malloc(sizeof(struct Node));//建立节点
p->data = s2[R2];//后序的最后一个节点就是根节点
for (i = L1; i < R1; i++) { //找到中序中的根节点所在位置
if (s1[i] == p->data)
break;
}
// 利用中序序列的相对位置确定后续序列的位置
p->Left = build_Tree(L1, i - 1, L2, (L2 + ((i - 1) - L1)));
p->Right = build_Tree(i + 1, R1, (R2 - (R1 - (i + 1))) - 1, (R2 - 1));
return p;
}
int best=100001, best_sum=10000000; //叶节点数据不超过10000
void dfs(TreeNode *p, int sum)
{
if (p == NULL)
return;
sum += p->data;
if (p->Left == NULL && p->Right == NULL) //如果是叶节点
{
if (sum < best_sum) //如果是符合条件的节点,则进行替换
{
best_sum = sum;
best = p->data;
}
else if (sum == best_sum) { //如果和权值相同 则判断叶节点
if (p->data < sum)
p->data = sum;
}
}
//递归调用左右节点
dfs(p->Left, sum);
dfs(p->Right, sum);
}
int check(int *data, char *s)//处理输入数据使用
{
int j = 0;
for (int i = 0; s[i] != '\0'; i++)
{
j++;
while (s[i] != ' ' && s[i] != '\0') {
data[j] = data[j] * 10 + s[i] - '0';//把节点数据变为整数
i++;
}
}
return j;
}
//主函数
int main()
{
int j = 0, n1, n2;
char s[100];
gets(s);//读入字符串
n1 = check(s1, s);
gets(s);
n2 = check(s2, s);
TreeNode *head = build_Tree(1, n1 - 1, 1, n2 - 1);
dfs(head, 0);
printf("%d", best);
return 0;
}