5439: 数据结构实验：平衡二叉树

5439: 数据结构实验：平衡二叉树

时间限制(普通/Java):1000MS/3000MS     内存限制:65536KByte
总提交:67           测试通过:18

描述

判断一棵二叉树是否是平衡二叉树。

平衡二叉搜索树（Self-balancing binary search tree）又被称为AVL树（有别于AVL算法），且具有以下性质：它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

二叉树节点定义如下：

struct TreeNode {

    int val;

    struct TreeNode *left;

    struct TreeNode *right;

};

题目部分代码已经完成，您只需要补充并提交以下函数：

int isBalanced(struct TreeNode *root);//如果是平衡二叉树，返回1，否则返回0

输入

输入为若干个整数（不超过512）表示一棵二叉树顺序表示时的结点元素值，其中0表示二叉树对应结点为空。输入以-1结束。 

输出

如果是平衡二叉树则输出Yes，否则输出No。

样例输入

1 2 3 0 0 4 5 -1

样例输出

Yes

题目来源

TOJ

#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;
}