码迷,mamicode.com
首页 > 其他好文 > 详细

【TOJ 5439】数据结构实验:平衡二叉树

时间:2018-05-21 14:38:35      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:int   判断   space   otto   ++   返回   std   mil   ace   

描述

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

平衡二叉搜索树(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

#include<bits/stdc++.h>
using namespace std; 
struct TreeNode{
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
};
struct TreeNode*creat()
{
    int front=1,rear=0,x;
    TreeNode*qu[1005],*t,*root=NULL;
    while(scanf("%d",&x)!=EOF,x!=-1)
    {
        if(x==0)
            t=NULL;
        else
        {
            t=(struct TreeNode*)malloc(sizeof(struct TreeNode));
            t->val=x;
            t->left=NULL;
            t->right=NULL;
        }
        qu[++rear]=t;
        if(rear==1)
            root=t;
        else 
        {
            if(t&&qu[front])
            {
                if(rear%2==0)
                    qu[front]->left=t;
                else
                    qu[front++]->right=t;
            }
        }
    }
    return root;
}
int flag=1;
int dfs(struct TreeNode *root)
{
    if(!root)
        return 0;
    int x=dfs(root->left)+1;
    int y=dfs(root->right)+1;
    if(abs(x-y)>1)
        flag=0;
    return x>y?x:y;
}
int isBalanced(struct TreeNode *root)
{
    dfs(root);
    return flag;
}
int main()
{
    TreeNode*t=creat();
    if(isBalanced(t))
        cout<<"Yes\n";
    else 
        cout<<"No\n";
    return 0;
}

 

 

【TOJ 5439】数据结构实验:平衡二叉树

标签:int   判断   space   otto   ++   返回   std   mil   ace   

原文地址:https://www.cnblogs.com/kannyi/p/9066614.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!