PAT 1110 Complete Binary Tree[比较]

最新推荐文章于 2022-05-14 21:25:06 发布

weixin_34236869

最新推荐文章于 2022-05-14 21:25:06 发布

阅读量138

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/BlueBlueSea/p/9939813.html

版权

1110 Complete Binary Tree （25 分）

Given a tree, you are supposed to tell if it is a complete binary tree.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer

Output Specification:

For each case, print in one line YES and the index of the last node if the tree is a complete binary tree, or NO and the index of the root if not. There must be exactly one space separating the word and the number.

Sample Input 1:

9
7 8
- -
- -
- -
0 1
2 3
4 5
- -
- -

Sample Output 1:

YES 8

Sample Input 2:

8
- -
4 5
0 6
- -
2 3
- 7
- -
- -

Sample Output 2:

NO 1

题目大意：给出一个二叉树，判断是否是完全二叉树。

//这个题真的学到了不少东西。我的AC：

参照了：https://blog.csdn.net/hyf20144055065/article/details/51970789

#include <iostream>
#include <algorithm>
#include<cstdio>
#include<stdio.h>
#include <queue>
#include<cmath>
#include <vector>
using namespace std;
struct Node{
    int left=-1,right=-1;
}node[20];
int isr[20];
int root=-1;
int ct=0,n,last=0;
void level(int r){//进行层次遍历。我是选择在入队的时候+，但是这样是比较复杂的，我应该在出队的时候+。
    queue<int> qu;
    qu.push(r);
    //ct++;
    while(!qu.empty()){
        //if(ct==n)last=qu.back();
        int top=qu.front();qu.pop();
        if(top==-1)break;
        last=top;//每次last都赋值为当前。
        ct++;
//        if(node[top].left==-1)break;
//        else {
//            qu.push(node[top].left);
//            ct++;
//        }
        qu.push(node[top].left);//其实这里-1完全可以push进去，因为下一次再次弹出时会进行判断的。
        qu.push(node[top].right);
//        if(node[top].left!=-1){
//            qu.push(node[top].right);
//            ct++;
//        }
    }
}

int main()
{
    cin>>n;
    string l,r;
    for(int i=0;i<n;i++){
        cin>>l>>r;
        if(l!="-"){
            node[i].left=atoi(l.c_str());//注意这里的转换
            //cout<<node[i].left<<'\n';
            isr[node[i].left]=1;
        }
        if(r!="-"){
            node[i].right=atoi(r.c_str());
            isr[node[i].right]=1;
        }
    }
    for(int i=0;i<n;i++){
        if(isr[i]==0){
            root=i;break;
        }
    }
    level(root);
    if(ct==n)
        cout<<"YES "<<last;
    else
        cout<<"NO "<<root;
    return 0;
}