SWUST OJ 1077: 平衡二叉树的判定 C++实现

题目描述

编写程序判断给定的二叉树是否是平衡二叉树。

输入

二叉树的先序序列。

输出

如果是平衡二叉树，输出yes！，否者输出no！

#include<bits/stdc++.h>
using namespace std;
struct{
	int l, r;
}z[105];
int ans = 1;
int creat(){
	char ch;
	cin>>ch;
	if(ch=='#') return ch;
	z[ch].l = creat();
	z[ch].r = creat();
	return ch;
}
int d(char c){
	if(c=='#') return 0;
	else return (1+max(d(z[c].l), d(z[c].r)));
}
void dfs(char c){
	if(c=='#') return;
	int l = d(z[c].l);
	int r = d(z[c].r);
	int h = abs(l-r);
	if(h>1) ans = 0;
	return;
}
int main(){
	creat();
    dfs('A');
    cout<<((ans==0)?"no!":"yes!");
	return 0;
}