5432: 数据结构实验：二叉树倾斜度

5432: 数据结构实验：二叉树倾斜度

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

描述

求一棵二叉树的倾斜度，其中一个节点的倾斜度定义为“左子树元素之和”与“右子树元素之和”差的绝对值（空节点的元素值为0），二叉树的倾斜度定义为所有节点的倾斜度之和。

二叉树节点定义如下：

struct TreeNode {

    int val;

    struct TreeNode *left;

    struct TreeNode *right;

};

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

int findTilt(struct TreeNode* root);

输入

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

输出

输出二叉树的倾斜度。

样例输入

1 2 3 -1

样例输出

1

题目来源

TOJ

#include <stdio.h>
#include <stdlib.h>

struct TreeNode {
    int val;
    struct TreeNode *left,*right;
};

struct TreeNode a[520];

struct TreeNode* Creat(struct TreeNode* root)
{
    int k=0,i,j,n;
    while(scanf("%d",&n),n!=-1)
        a[k++].val=n;
    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 sum=0;

void rfs(struct TreeNode* node,int *n)
{
    if(!node) return;
    *n+=node->val;
    rfs(node->left,n);
    rfs(node->right,n);
}

void dfs(struct TreeNode* node)
{
    if(!node) return;
    int n=0,m=0;
    rfs(node->left,&n);
    rfs(node->right,&m);
    sum+=abs(n-m);
    dfs(node->left);
    dfs(node->right);
}

int findTilt(struct TreeNode* root)
{
    dfs(root);
    return sum;
}

int main()
{
    struct TreeNode* root=NULL;
    root=Creat(root);
    printf("%d\n",findTilt(root));
    return 0;
}