Just oj 2018 C语言程序设计竞赛（高级组）H: CBT?（完全二叉树）

H: CBT?
时间限制: 1 s      内存限制: 128 MB      提交 我的状态
题目描述
对于二叉树，如果这棵树的节点排布是按行从上到下，每行从左到右挨个放置，中间不会有空闲的节点。则我们称之为完全二叉树。

注：这棵树的根节点的值一定是1

输入
输入数字正整数n （1≤n≤201≤n≤20）

接下来n行，每行为两个数字(a,b)和一个字符c(L 或者 R)，如果字符c是L，则表示b是a的左子节点；如果字符c是R，则表示b是a的右子节点。 (1≤a,b≤30001≤a,b≤3000)

输出
判断这棵树是否为完全二叉树，如果是则输出Yes，否则输出No

样例输入
5
1 2 L
1 3 R
2 4 L
2 5 R
3 6 L
样例输出
Yes

分析：建立一个类型P(u,k),u为该点的值，k用于选择该点的左右子结点；用map储存u点子节点的值，用队列实现从根节点开始往下再往右遍历的过程；通过改变k的值实现同一结点从左儿子到右儿子遍历的过程，左或右儿子不存在则退出循环，若通过遍历的结点数量验证是否为完全二叉树；

#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
typedef pair<int,char>M;//用于选择左节点和右结点
int x[3033];
bool vis[3033];//用于寻找根节点
map<M,int>p;//用于寻找子节点
queue<int>q;//用于遍历树
int main()
{
        int n,y;
        char c;
        scanf("%d",&n);
        M tmp;
        for(int i=0;i<n;i++){
                scanf("%d%d",&x[i],&y);
                cin>>c;
                tmp.first=x[i],tmp.second=c;
                p[tmp]=y;
                vis[y]=1;
        }
        for(int i=1;i<=n;i++)
                if(!vis[x[i]]){
                        q.push(x[i]);break;//插入根节点
                }
        int cnt=0;
        while(!q.empty()){
                tmp.first=q.front();
                q.pop();
                tmp.second='L';
                if(p.find(tmp)!=p.end()){
                        q.push(p[tmp]);//插入tmp的左儿子
                        cnt++;
                }else{
                        break;
                }
                tmp.second='R';
                if(p.find(tmp)!=p.end()){
                        q.push(p[tmp]);//插入tmp的左儿子
                        cnt++;
                }else{
                        break;
                }
        }
        cnt==n?printf("Yes\n"):printf("No\n");
        return 0;
}

或者用数组

#include<iostream>
#include<queue>
using namespace std;
int x[3033];
bool vis[3033];
int nod[3033][3];
queue<int>q;
int main()
{
        int n;
        scanf("%d",&n);
        for(int i=0;i<n;i++){
                int y;
                char c[3];
                scanf("%d%d%s",&x[i],&y,c);
                vis[y]=1;
                if(c[0]=='L')
                        nod[x[i]][0]=y;
                else
                        nod[x[i]][1]=y;
        }
        for(int i=0;i<n;i++)
                if(!vis[x[i]]){
                        q.push(x[i]);break;
                }
        int cnt=0;
        while(!q.empty())
        {
                int n=q.front();
                q.pop();
                if(nod[n][0])
                        q.push(nod[n][0]),cnt++;
                else
                        break;
                if(nod[n][1])
                        q.push(nod[n][1]),cnt++;
                else
                        break;
        }
        cnt==n?printf("Yes\n"):printf("No\n");
        return 0;
}