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

二叉树4(天平问题)

时间:2018-05-12 13:46:35      阅读:178      评论:0      收藏:0      [点我收藏+]

标签:std   ret   stream   create   tor   否则   space   格式   new   

二叉树的深度遍历

输入一个树状天平,根据力矩相等原则判断是否平衡。平衡输出YES,否则输出NO。所谓力矩相等,就是,其中和分别为左右两边砝码的重量,D为距离。

采用递归(先序)方式输入:每个天平的格式为,,,,当或为0时,表示该“砝码”实际是一个子天平,接下来会描述这个子天平。当时,会先描述左子天平,然后是右子天平。

 

例如输入:

0 2 0 4

0 3 0 1

1 1 1 1

2 4 4 2

1 6 3 2

输出:

YES

#include"iostream"
using namespace std;
struct Tree{
    int data,edge;
    Tree *left;
    Tree *right;
};
void tree(Tree* &t,int data,int edge){
    t = new Tree;
    t->data = data;
    t->edge = edge;
    t->left = NULL;
    t->right = NULL;
}
void createTree(Tree* &t){
    int datal,datar,edge;
    cin>>datal>>edge;
    tree(t->left,datal,edge);
    cin>>datar>>edge;
    tree(t->right,datar,edge);
    if(datal == 0){
        createTree(t->left);
    }
    if(datar == 0){
        createTree(t->right);
    }
}
int torque(Tree *t){
    if(t->right==NULL){
        return t->data;
    }
    else if(t){
        int l = torque(t->left);
        int r = torque(t->right);
        if(l && r && r * t->right->edge == l * t->left->edge){
            return r + l;
        }
    }
    return 0;
}
int main(){
    Tree *t = new Tree;
    tree(t,0,0);
    createTree(t);
    if(torque(t)){
        cout<<"YES"<<endl;
    }
    else{
        cout<<"NO"<<endl;
    }
    return 0;
}

/*
0 2 0 4
0 3 0 1
1 1 1 1
2 4 4 2
1 6 3 2

*/

 

二叉树4(天平问题)

标签:std   ret   stream   create   tor   否则   space   格式   new   

原文地址:https://www.cnblogs.com/oleolema/p/9028447.html

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