toj2801 Binary Trees

来源:互联网 发布:js 数组比大小 编辑:程序博客网 时间:2024/06/05 08:08

题目链接:http://acm.tju.edu.cn/toj/showp.php?pid=2801

题目大意:给定一个字符串的后序遍历 求中序遍历 大写字母表示非叶子节点 小写字母表示叶子节点。

思路:后序遍历的最后一个为根,所以每次找出根,再分出左右子树,同样,左右子树的最后一个节点也是根。关键就是再找出根后怎么把左右子树找出来,因为是完全二叉树,所以子树中小写字母一定比大写字母多1。直接套树遍历的递归模板即可。

代码如下:

#include <iostream>
using namespace std;
string s;

void solve(int head, int end)
{
     if(head==end)
     {
        cout<<s[head];
        return;
     }
     else
     {
         int t;
         int count = 0;
         for(int i=end-1;i>=head;i--)  //循环没有算最后一个节点end
         {
             if(s[i]>='A'&&s[i]<='Z')
             count--;
             else
             count++;
             if(count == 1)
             {
               t =i;      //左右子树的分界点
               break;
             }
         }
         solve(head,t-1);  //左根右:中序遍历
         cout<<s[end];
         solve(t,end-1);
     }
}

int main()
{
    int num;
    cin>>num;
    while(num--)
    {
      cin>>s;
      solve(0,s.length()-1);
      cout<<endl;
    }
return 0;
}