1110. Complete Binary Tree (25)

转载于 2017-08-01 14:35:00 发布 · 50 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wsggb123/p/7268436.html

本文介绍了一个使用C++实现的基于广度优先搜索（BFS）的二叉树遍历算法。该算法能够正确判断输入的二叉树是否完全遍历，并输出最后访问到的节点编号。文章通过实例演示了如何读取二叉树结构并进行BFS遍历的过程。

#include<iostream>
#include<string>
#include<map>
#include<vector>
#include<algorithm>
#include<queue>
#include<set>
#include<stack>
using namespace std;

struct node {
    int left=-1, right=-1;
    int val;
};
int num;
int cnt[10001];
node arr[21];
int c = 0;
int last;
bool f;
void bfs(int root) {

    queue<int> q;
    q.push(root);

    while (!q.empty()) {
        int temp = q.front();
        

        if (temp == -1) {
            //cout << c;
            if (c == num) {
                f = 1;
            }

            else
                f = 0; 
            break;


        }

        q.pop();
        c++;
        q.push(arr[temp].left);
        q.push(arr[temp].right);
        last = temp;

    }

}




int main() {
    
    
    cin >> num;
    

    for (int i = 0; i < num; i++) {
        arr[i].val = i;
        string l, r;
        cin >> l >> r;
        if (l != "-") {
            arr[i].left = stoi(l);
            cnt[stoi(l)] = 1;
        }
            
        if (r != "-") {
            arr[i].right = stoi(r);
            cnt[stoi(r)] = 1;
        }
            
    }

    int root = 0;
    while (cnt[root] == 1) {
        root++;
    }
    
    bfs(root);
    //cout << f;

    if (f == 1) {
        cout << "YES" << ' ' << last;
    }

    else {
        cout << "NO" << ' ' << root;
    }
    system("pause");
    
}