天平问题 Not so Mobile UVa 839

天平 Not so Mobile UVa 839

题述
输入一个树状天平，根据力矩相等原则判断 是否平衡。如图所示，所谓力矩相等就是WlDl=WrDr,其中Wl和Wr分别为左右两边砝码的重量，D为距离。采用递归(先序)的方式输入:每个天平的格式为Wl，Dl，Wr，Dr，当Wl或Wr为0时，表示该“砝码”实际上是一个子天平，接下来会描述这个子天平。当Wl=Wr=0时，会先描述左子天平，然后是右子天平。
样例输入
1
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;

bool solve(int &W){
    int W1,D1,W2,D2;
    bool b1=true,b2=true; //初始化默认左右天平都平衡
    cin>>W1>>D1>>W2>>D2;
    if(!W1)
        b1=solve(W1);  //判断左天平是否平衡
    if(!W2)
        b2=solve(W2); //判断右天平是否平衡
    W=W1+W2;
    return b1&&b2&&(W1*D1==W2*D2);
}

int main()
{
    int T,W;    //W用来存储天平总质量,这道题其实没用到，但是如果有这样的要求，这道题就给了思路
    cin>>T;
    while(T--){
        if(solve(W))
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
        if(T)
            cout<<endl;
    }
    return 0;
}