5439: 数据结构实验:平衡二叉树
时间限制(普通/Java):1000MS/3000MS 内存限制:65536KByte
总提交:67 测试通过:18
总提交:67 测试通过:18
描述
判断一棵二叉树是否是平衡二叉树。
平衡二叉搜索树(Self-balancing binary search tree)又被称为AVL树(有别于AVL算法),且具有以下性质:它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
二叉树节点定义如下:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
扫描二维码关注公众号,回复:
1498184 查看本文章
};
题目部分代码已经完成,您只需要补充并提交以下函数:
int isBalanced(struct TreeNode *root);//如果是平衡二叉树,返回1,否则返回0
输入
输入为若干个整数(不超过512)表示一棵二叉树顺序表示时的结点元素值,其中0表示二叉树对应结点为空。输入以-1结束。
输出
如果是平衡二叉树则输出Yes,否则输出No。
样例输入
1 2 3 0 0 4 5 -1
样例输出
Yes
题目来源
#include <stdio.h> #include <stdlib.h> struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; }; struct TreeNode a[520]; int flag=1; struct TreeNode* Creat() { int x,k=0,i,j; while(scanf("%d",&x),x!=-1) a[k++].val=x; for(i=0,j=1;i<k;i++,j++) { if(i+j>=k||a[i+j].val==0) a[i].left=NULL; else a[i].left=&a[i+j]; if(i+j+1>=k||a[i+j+1].val==0) a[i].right=NULL; else a[i].right=&a[i+j+1]; } return &a[0]; }; int depth(struct TreeNode *root) { if(!root) return 0; int ls=depth(root->left); int rs=depth(root->right); return ls>rs?(ls+1):(rs+1); } void dfs(struct TreeNode *root) { if(root) { int ls=depth(root->left); int rs=depth(root->right); if(abs(ls-rs)>1) flag=0; else { dfs(root->left); dfs(root->right); } } return; } int isBalanced(struct TreeNode *root) { dfs(root); return flag; } int main() { struct TreeNode *root=Creat(); if(isBalanced(root)==1) printf("Yes\n"); else printf("No\n"); return 0; }