题目内容
具体题目见连接
https://www.patest.cn/contests/pat-a-practise/1110
判断完全二叉树
思路借助队列,层次分析法的改进。当遇到节点没有左孩子或者没有右孩子,就退出,通常在判断左右孩子的时候,统计几点的数量。因为完全二叉树的概念是除了最后一层和倒数第二层,其余的各层的每个节点都有左孩子和右孩子。因此可以这么用。
代码如下
#include <iostream>
#include <queue>
#include <stdio.h>
#include <sstream>
#include <stdlib.h>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
using namespace std;
struct BiNode{
int lchild,rchild,data;
};
const int NUM =25;
BiNode BiTree[NUM];
int num[NUM]={0};
int levelSequence[NUM]={-1};
void InOrder(int root){
if(BiTree[root].lchild!=-1){
InOrder(BiTree[root].lchild);
}
cout<<BiTree[root].data;
if(BiTree[root].rchild!=-1){
InOrder(BiTree[root].rchild);
}
}
void PreOrder(int root){
cout<<BiTree[root].data;
if(BiTree[root].lchild!=-1){
PreOrder(BiTree[root].lchild);
}
if(BiTree[root].rchild!=-1){
PreOrder(BiTree[root].rchild);
}
}
void outPut(int levelSequence[],int index){
for(int i=0;i<index;i++){
cout<<levelSequence[i]<<" ";
}
cout<<endl;
}
int stringtoint(string s){
stringstream ss;
ss << s;
int n;
ss >> n;
return n;
}
int main(int argc, char *argv[]) {
//int n;
// cin>>n;
// //建立二叉树
// for(int i=0;i<n;i++){
// num[i] = 0;
// int lchild,rchild;
// string left,right;
// cin>>left>>right;
// if(left.substr(0,1)=="-"){
// lchild = -1;
// }
// else{
// lchild = stringtoint(left);
// num[lchild]++;
// }
// if(right.substr(0,1)=="-"){
// rchild = -1;
// }
// else{
// rchild = stringtoint(right);
// num[rchild]++;
// }
// //cout<<lchild<<" "<<rchild<<endl;
// BiTree[i].lchild = lchild;
// BiTree[i].rchild = rchild;
// BiTree[i].data = i;
// }
int N;
scanf("%d", &N);
char l[3], r[3];
for (int i = 0; i < N; i++)
{
scanf("%s %s", l, r);//注意将之前的换行符吸收掉
//printf("%s %s",l,r);
BiTree[i].data = i;//注意下标可能是两位数,不能只用一个字符储存!!!!
if (l[0] != '-')
{
BiTree[i].lchild = atoi(l);
//cout<<BiTree[i].lchild<<endl;
//notroot[atoi(l)] = true;
num[BiTree[i].lchild]++;
}
else{
BiTree[i].lchild = -1;
}
if (r[0] != '-')
{
BiTree[i].rchild = atoi(r);
//cout<<BiTree[i].rchild<<endl;
//notroot[atoi(r)] = true;
num[BiTree[i].rchild]++;
}
else{
BiTree[i].rchild = -1;
}
}
int root;
//找根节点
for(int i=0;i<N;i++){
if(num[i]==0){
root = i;
break;
}
}
//输出二叉树中序遍历
//InOrder(root);
//输出前序遍历
//PreOrder(root);
//真层次遍历
int index = 0;
queue<int> level;
level.push(root);
while(!level.empty()){
int temp = level.front();
levelSequence[index++] = temp;
//cout<<temp;
level.pop();
if(BiTree[temp].lchild!=-1){
level.push(BiTree[temp].lchild);
}
if(BiTree[temp].rchild!=-1){
level.push(BiTree[temp].rchild);
}
}
//输出层次遍历
//outPut(levelSequence,index);
//层次遍历,根据节点个数判断是不是二叉树
queue<int> q;
q.push(root);
int sum = 1;
int last =root;
int top=-1;
while(!q.empty()){
top=q.front();
q.pop();
if(BiTree[top].lchild!=-1)
{
sum++;
q.push(BiTree[top].lchild);
last = BiTree[top].lchild;
}
else{
break;
}
if(BiTree[top].rchild!=-1)
{
sum++;
q.push(BiTree[top].rchild);
last = BiTree[top].rchild;
}
else{
break;
}
}
if(sum==N)
printf("YES %d\n",last);
else
printf("NO %d\n",root);//这个top保留了最后一个节点的父亲节点
return 0;
}
代码说明
1.代码中不仅有判断二叉树,在注释中也有层次遍历,前序中序遍历等。可以参考看看。
2.为什么我在输入的时候用string,就老是出错,出错吧!就是注释那部分,请指正!