poj3437
来源:互联网 发布:adobe encoder mac 编辑:程序博客网 时间:2024/05/22 07:09
这道题当时看的眼都花了,明白之后还是没有什么思路,最后看别人的代码才算ac
思路是这样的:对于读入的字符串,d表示下一层结点,计算出每个节点在二叉树 节点的层次,如果这个节点层次大于之前二叉树的高度,那么更新二叉树即可,重点在于处理好 d 字符
拿样例1 来说,第一个 d 时 遍历结点1 之后 up ,而结点 1 在二叉树中的层次为 1 ,这时可以把二叉树高度更新为1,u之后返回0结点,继续处理2 结点,由于2 结点在二叉树中为1 结点的右儿子,所以2 结点的层次 为2,实际上可以得出这样一个结论,当由 u 向上返回时,如果后面有一个 d 则这个节点 比 u 之前紧接的d 所对应 的结点在二叉树中的层次要+1,如果不明白可以好好看看样例。所以 1 为 第一层 ,2 为第二层 3是第三层。
- #include <iostream>
- #include <stdio.h>
- using namespace std;
- int cal_height(int level,int* height1)
- //level用来修改原树高度height1, 函数返回值为转换后的树高度
- {
- char c;
- int n;//n为当前结点子节点个数
- int height = 0;//转换后树高度
- if(level > *height1)//当前层次已经大于树高,则更新原树高度
- *height1 = level;
- int tmp;
- for(n = 0;(c = getchar() )=='d';n++)//由于兄弟节点转换后为右儿子节点,所以用n++,完成树高递增
- //目前d的个数为n+1,即该节点位于该子树的n+1层
- {
- if( (tmp = cal_height(level + 1,height1) + n + 1) > height)
- height = tmp;
- }
- return height;
- }
- int main()
- {
- freopen("graft.in","r",stdin);
- freopen("out.txt","w",stdout);
- int height1,height2,casenum = 1;
- char c;
- while((c = getchar()) != '#')
- {
- ungetc(c,stdin);
- height1 = 0;
- height2 = cal_height(0,&height1);
- printf("Tree %d: %d => %d\n",casenum++,height1,height2);
- }
- return 0;
- }