题目:输入一颗二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1,那么它就是一颗平衡二叉树。
分析:后续遍历,从二叉树的叶子节点开始一步一步的向上比较,一旦某个节点不满足平衡二叉树要求,则直接返回false,感觉有点像斐波那契的非递归算法。
#include <stdio.h>
#include <malloc.h>
typedef struct node
{
int value;
struct node *left;
struct node *right;
}node,*pnode;
void create_tree(pnode *root)//按照前序遍历的顺序输入
{
int value;
scanf("%d",&value);//输入值是要保证每个结点都有两个子结点,如果没有子结点则用-1表示
if(value == -1)
*root = NULL;
else
{
*root = (pnode)malloc(sizeof(node));
(*root)->value = value;
create_tree(&(*root)->left);
create_tree(&(*root)->right);
}
}
void print_tree(pnode root)
{
if(root==NULL)
return;
printf("%d\t",root->value);
print_tree(root->left);
print_tree(root->right);
}
bool isbalance_core(pnode root, int *depth)//后序遍历,先从下面开始比较,逐渐往上递增
{
if(root == NULL)//叶子节点的子结点
{
*depth = 0;
return true;
}
int left,right;
if(isbalance_core(root->left,&left) && isbalance_core(root->right,&right))
{
if(left-right<=1 && left-right>=-1)
{
*depth = 1+(left>right?left:right);
return true;
}
}
return false;
}
bool isbalance(pnode root)
{
if(root==NULL)
return false;
int depth;
return isbalance_core(root,&depth);
}
int main()
{
pnode root;
printf("请输入二叉树的结点值:\n");
create_tree(&root);
printf("所创建二叉树的前序遍历为:\n");
print_tree(root);
printf("\n");
if(isbalance(root))
printf("此二叉树是平衡二叉树!\n");
else
printf("此二叉树不是平衡二叉树!\n");
return 0;
}