题目
答案
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
typedef struct {
int key1;
int key2;
int left;
int right;
} Node;//创建结构体
//准备阶段
Node node[1010];//创建结构体数组用以储存输入的节点
int isSun[1010];//判断是否为子节点,1是,2不是
vector<int> vec;//创建vector储存中序遍历的结果 (vector可以理解为数组)
//bfs按层遍历每个节点是否符合小顶堆
bool bfs( int root ) {
queue<Node> q;//创建临时队列
q.push( node[root] );//将根节点放入队列
bool flag = true;
while( !q.empty() ) { //循环
if( !flag ) break;
Node curNode = q.front();
q.pop();
int left = curNode.left;
int right = curNode.right;
if( left != -1 && curNode.key2 >= node[left].key2 ) {//左节点不为空且左节点要key2比根节点大,否则为false
flag = false;
}
if( right != -1 && curNode.key2 >= node[right].key2 ) {//右节点不为空且右节点key2要比根节点大,否则为false
flag = false;
}
if( left != -1 ) q.push( node[left] );
if( right != -1 ) q.push( node[right] );
}
return flag;
}
//中序遍历node数组并将结果存入vec中
void dfsIn( int root ) {
if( node[root].left != -1 ) dfsIn( node[root].left );
vec.push_back( node[root].key1 );
if( node[root].right != -1 ) dfsIn( node[root].right );
}
int main() {
int n;
scanf( "%d", &n );
//遍历存储每个节点,同时如果节点是别的节点的左子树或右子树,那么它就一定不是根节点
for( int i = 0; i < n; i++ ) {
scanf( "%d%d%d%d", &node[i].key1, &node[i].key2,
&node[i].left, &node[i].right );
if( node[i].left != -1 ) {
isSun[node[i].left] = 1;
}
if( node[i].right != -1 ) {
isSun[node[i].right] = 1;
}
}
//遍历寻找根节点
int root = -1;
for( int i = 0; i < n; i++ ) {
if( isSun[i] == 0 ) {
root = i;
break;
}
}
// 判断空树
if( root == -1 ) {
printf( "YES\n" );
return 0;
}
bool flag = bfs( root );//获取bfs的结果
dfsIn( root );//进行中序遍历
//判断vec内的数字是否满足升序排列
bool temp = true;
for( int i = 0; i < n - 1; i++ ) {
if( vec[i] > vec[i + 1] ) {
temp = false;
break;
}
}
if( flag && temp ) printf( "YES" );
else printf( "NO" );
return 0;
}
总结
这道题分别采用bfs按层遍历以及中序遍历判断key2和key1是否满足条件,详细情况见注释