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;
}