首页 > 学院 > 开发设计 > 正文

CODE[vs] 天梯 1501 二叉树最大宽度和高度

2019-11-11 05:33:02
字体:
来源:转载
供稿:网友

二叉树最大宽度和高度 时间限制: 1 s 空间限制: 128000 KB 题目等级 : 白银 Silver 题解

题目描述 Description 给出一个二叉树,输出它的最大宽度和高度。

输入描述 Input Description 第一行一个整数n。

下面n行每行有两个数,对于第i行的两个数,代表编号为i的节点所连接的两个左右儿子的编号。如果没有某个儿子为空,则为0。

输出描述 Output Description 输出共一行,输出二叉树的最大宽度和高度,用一个空格隔开。

样例输入 Sample Input 5

2 3

4 5

0 0

0 0

0 0

样例输出 Sample Output 2 3

数据范围及提示 Data Size & Hint n<16

默认第一个是根节点

以输入的次序为编号

2-N+1行指的是这个节点的左孩子和右孩子

注意:第二题有极端数据!

1 0 0

这题你们别想投机取巧了,给我老老实实搜索!

思路:没啥思路,我就是造出一棵树然后dfs搜索,求出宽度和高度。ps:说实话,这个宽度我搞了好久,写重复了一个,弄的答案错了!

代码:(略长)

#include<iostream>#include<string.h>#include<math.h>#include<algorithm>using namespace std;struct tree{//树 int num; struct tree *leftT; struct tree *rightT;};struct tree *p;//指向当前需要的树节点void find(struct tree *tr, int i){ if (tr->num == i){ p = tr; return; } if (tr->leftT)find(tr->leftT, i); if (tr->rightT)find(tr->rightT, i);}//计算深度 int deepth(struct tree *tr){ if (!tr)return 0; int m = max(deepth(tr->leftT)+1, deepth(tr->rightT) + 1); return m;}int a[100], maxs = 1;//默认maxs的宽度为1,为根节点//计算宽度void width(struct tree * tr, int i){ if (!tr)return; if (tr->leftT)a[i + 1]++; if (tr->rightT)a[i + 1]++; if (maxs<a[i + 1])maxs = a[i + 1]; width(tr->leftT, i + 1); width(tr->rightT, i + 1);}void dfs(struct tree *tr, int i){ int n, m; cin >> n >> m; if (n == 0 && m == 0)return; find(tr, i); if (n > 0){ p->leftT = new tree; p->leftT->num = n; p->leftT->leftT = NULL; p->leftT->rightT = NULL; } if (m > 0){ p->rightT = new tree; p->rightT->num = m; p->rightT->leftT = NULL; p->rightT->rightT = NULL; }}int main(){ a[1] = 1; int n; cin >> n; struct tree *tr = new tree; tr->num = 1; tr->leftT = NULL; tr->rightT = NULL; for (int i = 1; i <= n; i++){ dfs(tr, i); } int deep = deepth(tr); width(tr, 1); cout << maxs << " " << deep << endl; return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表