(UVA - 839) Not so Mobile（递归+引用，树的先序遍历）

链接：https://vjudge.net/problem/UVA-839

题意：输入一个树状天平，根据力矩规则判断是否平衡。 Wl*Dl=Wr*Dr 先序遍历输入，每个天平的格式:Wl,Dl,Wr,Dr Wl或Wr=0时，表示输入一个子天平，接下来描述这个天平，当两者都为0，左右天平依次描述。

分析：直接在输入的时候判断子天平是否平衡，并更新子天平的重量（此处用到 了 C++中的引用） 紫书P157

#include<cstdio>
#include<set>
#include<cstring>
#include<sstream>
#include<iostream>
#include<algorithm>
using namespace std;
#define mem(a,n) memset(a,n,sizeof(a))
typedef long long LL;

const int N=25;
const int INF=0x3f3f3f3f;
#define same(x,y) Find(x)==Find(y)

bool solve(int& w)///此处的引用 可以起到修改左右子天平的总重量
{
    int w1,w2,d1,d2;
    int num1=0,num2=0;
    bool b1=true,b2=true;
    scanf("%d%d%d%d",&w1,&d1,&w2,&d2);
    if(!w1) num1++,b1=solve(w1);
  ///  printf("num1=%d b1=%d\n",num1,b1);
    if(!w2) num2++,b2=solve(w2);
 ///   printf("num2=%d b2=%d\n",num2,b2);
    w=w1+w2;
  ///  printf("w=%d\n",w);
    return b1&&b2&&(w1*d1==w2*d2);
}
int main()
{
    int t,w;
    scanf("%d",&t);
    while(t--)
    {
        printf("%s\n",solve(w)?"YES":"NO");
        if(t) printf("\n");
    }
    return 0;
}