Not so Mobile（树状天平，二叉树+递归）

最新推荐文章于 2024-10-04 13:53:41 发布

合金Bunny酱

最新推荐文章于 2024-10-04 13:53:41 发布

阅读量110

点赞数

分类专栏：小白的紫书刷题日记文章标签：算法

本文链接：https://blog.csdn.net/bunny_1024/article/details/121185881

版权

小白的紫书刷题日记专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何解决一个在线编程题目，涉及天平的平衡判断。通过自底向上的方法，递归计算子树的等效重量，并依据给定公式判断每个子天平及整体是否平衡。代码中展示了如何实现这一过程，输入为子天平的重量和距离，输出为YES或NO，表示天平是否平衡。

摘要由CSDN通过智能技术生成

题目链接：

Online JudgeOnline Judgehttps://onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=780

思路：

自底向上，依次判断各个子天平是否平衡，对于父天平，要判断是否平衡可以用子树的“等效重量”来计算（题目也已经给出了公式）

代码：

#include <iostream>
using namespace std;

bool solve(int &w) {
    int w1, d1, w2, d2;
    bool b1 = true, b2 = true; // 判断左右子树是否平衡的边浪
    cin >> w1 >> d1 >> w2 >> d2;
    if(w1 == 0) b1 = solve(w1); // 递归的计算w1的等效值
    if(w2 == 0) b2 = solve(w2);
    w = w1 + w2; // 当前子天平的等效重量
    return b1 && b2 && (w1*d1 == w2*d2); // 子树和当前树都平衡
}
int main(){
    int n; // 有n组数据
    cin >> n;
    int tmp = 0;
    while(n--){
        if(solve(tmp)) cout << "YES\n";
        else cout << "NO\n";
        if(n) cout << "\n"; // 题目格式要求
    }
    return 0;
};