UVA839 天平 Not so Mobile

这道题很难，除了题目本身，还有场外因素，那就是刘汝佳写了一个非常短且巧妙的答案，然后还说继续阅读前务必理解以下程序，但是我先不管它，用自己的理解也就是笨方法做，现在照抄刘汝佳的题解也没有意义，

首先这是一个二叉树的模型，那么结点储存的是4个值，也就是题目给的，每个结点理解为一个天平，这道题平衡的条件就是所有的结点都平衡，然后每个结点有左重和右重，左重就是输入时结点的左重和它所有的左边的子孙的重量之和，右重类似，这道题的输入是递归建树，我们易得左重不为零相当于这个结点没有左孩子，左重为零，相当于这个结点还有左孩子，需要向左边递归，由此可以写出建树，然后就是递归，这要设置一个全局变量，来判断，这个函数用来求当前结点的重量，返回给上一层，实际重量=输入重量+子孙重量，我们这个返回的就是子孙重量，加上输入的也就是二叉树里面保存的重量，组成了实际重量，就可以判断当前的天平是不是平衡的，

我一开始理解为只需要判断根结点是不是平衡的，写的代码就是判断一个，结果提交uDebug里面666个数据量的样例都过了，说明那个样例666个全是只考虑根节点的，有点坑

#include <bits/stdc++.h>

#define fi first
#define se second
#define pb push_back
#define mk make_pair
#define sz(x) ((int) (x).size())
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;

const int N = 1e4 + 5;

struct Node {
    int wl, dl, wr, dr;
    Node *lchild, *rchild;
};

int ok;

Node *build() {
    int wl, dl, wr, dr;
    cin >> wl >> dl >> wr >> dr;
    Node *root = new Node;
    root->wl = wl; root->dl = dl;
    root->wr = wr; root->dr = dr;
    root->lchild = (wl ? NULL : build());
    root->rchild = (wr ? NULL : build());
    return root;
}

int judge(Node *root, int level) {
    int wl = root->wl;
    int dl = root->dl;
    int wr = root->wr;
    int dr = root->dr;
    if (wl && wr) {
        if (wl * dl != wr * dr) ok = 0;
        return wl + wr;
    } else if (wl) {
        int t = judge(root->rchild, level + 1);
        if (wl * dl != t * dr) ok = 0;
        return wl + t;
    } else if (wr) {
        int t = judge(root->lchild, level + 1);
        if (t * dl != wr * dr) ok = 0;
        return t + wr;
    } else {
        int a = judge(root->lchild, level + 1);
        int b = judge(root->rchild, level + 1);
        if (a * dl != b * dr) ok = 0;
        return a + b;
    }
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        Node *root = build();
        ok = 1;
        int tt = judge(root, 1);
        cout << (ok ? "YES" : "NO") << endl;
        if (t) cout << endl;
    }
    return 0;
}