UVA 122 Trees on the level 二叉树

题目：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=19630

题意：给一些形如(n,s)的输入，n是数字，代表值，s是字符串，代表在二叉树中的位置。然后输出二叉树的层次遍历序列，若信息不全或节点有重复，输出not complete
思路：我的思路是可以对位置字符串进行排序，观察可知长度短的一定在上面，并且L的字典序小于R，于是先按长度排序，长度相等的按照字典序排序，排好后就是二叉树的层次序列。但是我们要判断二叉树是否信息不全，即判断根节点是否存在和每个点是否有父节点，还要判断每个点不能有重复，把位置字符串末尾去掉一位就是父节点的位置，判断一下即可（因为根节点的表示原因，第二层要特判），判重就更简单了。另外讲一下我同学的思路：他的思路是直接建二叉树，比如（7，LLL），建左 左 左三个节点，但只给最后一个赋值为7，那两个先不管，等到后来遇到再赋值。建好二叉树后遍历判断

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
#include <map>
#include <string>
#include <vector>
#include <functional>
#define bug puts("here")
using namespace std;

const int N = 1010;
struct node //开结构体记录每个节点的值和位置
{
    string s;
    int v;
}g[N];
map<string, int> mpa;//储存所有的位置字符串，可以用来查询某个位置是否存在，用set更合理||-_-
bool cmp(const node &a, const node &b) //对位置按二叉树层次排序，首先按长度排序，然后按字典序
{                                      //因为长度短的一定在上面，而L的字典序比R小，因此这样排
    if(a.s.size() == b.s.size())
        return a.s < b.s;
    else return a.s.size() < b.s.size();
}
void chu(char *s, int cnt)
{
    int i;
    int sum = 0;
    for(i = 1; s[i] != ','; i++) //分离括号中的数字
        sum = sum * 10 + s[i] - '0';
    char str[N];
    memset(str, 0, sizeof str);
    int k = 0;
    for(i++; s[i] != ')'; i++) //分离代表位置的字符串
        str[k++] = s[i];
    if(str[0] == '\0') //存到数组中，对于根节点我用了"A"表示
    {
        g[cnt].s = "A";
        g[cnt].v = sum;
        mpa["A"] = sum;
    }
    else //非根节点
    {
        g[cnt].s = str;
        g[cnt].v = sum;
        mpa[str] = sum;
    }
}
int main()
{
    char s[N];
    while(~ scanf("%s", s))
    {
        int cnt = 0;
        mpa.clear();
        if(strcmp(s, "()") == 0)
        {
            printf("not complete\n");
            continue;
        }
        chu(s, cnt++);
        while(scanf("%s", s), strcmp(s, "()"))
            chu(s, cnt++);
        bool flag = true;
        int res[N];
        int k = 0;
        sort(g, g + cnt, cmp);
        for(int i = 1; i < cnt; i++) //有没有重复的节点，若有直接输出not complete
            if(g[i].s == g[i-1].s)
            {
                flag = false;
                break;
            }
        if(flag == false)
        {
            printf("not complete\n");
            continue;
        }
        for(int i = 0; i < cnt; i++)//查询每个点的父节点是否存在，不存在说明信息不完全
        {
            if(k == 0) //判断第一个点是不是根节点
            {
                if(g[i].s == "A") res[k++] = g[i].v;
                else
                {
                    flag = false;
                    break;
                }
            }
            else //把每个点表示位置的字符串去掉最后一位，就是父节点的位置，第二层除外，所以第二层特判，因为我用A表示的根节点
            {
                string s = g[i].s;
                int len = s.size();
                if(len == 1) 
                    res[k++] = g[i].v;
                else
                {
                    s[len-1] = '\0';
                    if(mpa.find(s) == mpa.end())
                        res[k++] = g[i].v;
                    else
                    {
                        flag = false;
                        break;
                    }
                }
            }
        }
        if(flag)
            for(int i = 0; i < k; i++)
                printf("%d%c", res[i], i == k - 1 ? '\n' : ' ');
        else printf("not complete\n");
    }
    return 0;
}