7-1 二叉树最长路径 (100 分)
给定一棵二叉树T,求T中的最长路径的长度,并输出此路径上各结点的值。若有多条最长路径,输出最右侧的那条。
输入格式:
第1行,1个整数n,表示二叉树有n个结点, 1≤n≤100000.
第2行,2n+1个整数,用空格分隔,表示T的扩展先根序列, -1表示空指针,结点用编号1到n表示。
输出格式:
第1行,1个整数length,length表示T中的最长路径的长度。
第2行,length+1个整数,用空格分隔,表示最右侧的最长路径。
输入样例:
在这里给出一组输入。例如:
5
1 2 -1 -1 3 4 -1 -1 5 -1 -1
输出样例:
在这里给出相应的输出。例如:
2
1 3 5
思路
先建树,再按题目优先级要求先右后左遍历
这里递归建树 ,由于给出先根序列,就先创建当前结点,再把左右指针弄出来,注意返回一个“地址”,方便回溯到上一层时赋给相应儿子指针
这里以叶结点为根走不出路径(初始lenth=0),dfs(x)代表以x为根的题中所述最长路径(问题分解成同构子问题)
dfs默认走右(优先级),除非左子树最长路径大,注意dfs的回溯可以看作沿父子边反向回去,故lenth更新为dfs(子树的根)+1
最后输出dfs(题中树的根),并按方向数组fs走把路径打出来
代码
#include<iostream>
#include<cstdio>
using namespace std;
const int inf=205500;
int a[inf];
int tot=1;
int cnt=1;
int num[inf];
int lef[inf];
int righ[inf];
int maxlenth=0;
bool fs[inf];
int make_tree()
{
if(a[cnt]==-1)
{
cnt++;
return -1;
}//创建节点
int dq=tot;
tot++;
num[dq]=a[cnt++];//data域
lef[dq]=make_tree();//左指针
righ[dq]=make_tree();//右指针
return dq;//返回地址,回溯时赋值
}
int dfs(int id)
{
int lenth=0;
fs[id]=1;//默认右
int dq;
if(righ[id]!=-1)
{
dq=dfs(righ[id])+1;//右子树路径长(回溯相当于多一条边)
if(dq>lenth)
{
lenth=d