Trees on the level UVA - 122

这道题设计到二叉树的层次遍历,直接开数组暴力去装肯定装不下,所以我们可以人为的按照输入顺序对每一个节点进行标号,然后用r[l]表示该节点的右子树,l[i]表示该节点的左子树,它们的值为-1时表示没有左子树或右子树.

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=300;
int r[N],l[N],val[N];
vector<int>G;
int main()
{
    string s;
    while(cin>>s)
    {
        memset(r,-1,sizeof r);
        memset(l,-1,sizeof l);
        memset(val,-1,sizeof val);
        bool flag=1;
        int num1=0,cnt=1;
        while(s!="()")
        {
            num1++;
            int i,len=s.size(),num=0;
            for(i=1;i<len;i++)
            {
                int v=int (s[i]-'0');
                if(v<0||v>9)
                    break;
                num=num*10+v;
            }
            string s2=s.substr(i+1,len-i-2);
            if(s2.size()==0)
            {
                if(val[1]!=-1)
                    flag=0;
                val[1]=num;
                cin>>s;
                continue;
            }
            int u=1;
            for(int j=0;j<s2.size();j++)
            {
                if(s2[j]=='L')
                {
                     if(l[u]==-1)
                        l[u]=++cnt;//创建一个新的左孩子
                     u=l[u];
                }
                else
                {
                    if(r[u]==-1)
                        r[u]=++cnt;创建一个新的右孩子
                    u=r[u];
                }
            }
            if(val[u]!=-1)//该节点已被赋值
                flag=0;
            val[u]=num;
            cin>>s;
        }
        queue<int>q;
        q.push(1);
        G.clear();
        while(!q.empty())
        {
            int aa=q.front();
            q.pop();
            if(val[aa]==-1) {flag=0;break;}//该节点被路过但没有被赋值
            G.push_back(val[aa]);
            if(l[aa]!=-1) q.push(l[aa]);
            if(r[aa]!=-1) q.push(r[aa]);
        }
        if(G.size()==num1&&flag)
        {
            cout<<G[0];
            for(int i=1;i<G.size();i++)
                cout<<" "<<G[i];
            cout<<endl;
        }
        else
            cout<<"not complete"<<endl;
    }
    return 0;
}