UVA 839 Not so Mobile (递归建树 判断)

Posted 2020-06-23

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
#include<string>
#include<math.h>
#include<stack>
#include<cstdlib>
#include<algorithm>
#include<cctype>
#include<sstream>

using namespace std; 

bool solve(int& W){
    int l,w1,r,w2;
    bool b1 = true,b2 = true; //判断左右是否满足 
    cin >> w1>> l >> w2 >> r;    
    if(!w1) b1 = solve(w1);    //重量为0,则继续递归 
    if(!w2) b2 = solve(w2);    //递归 
     W = w1 + w2; //是父亲节点 w1 为0时,所有子树的权重之和 W 返回给调用者 w1 
     return b1 && b2 && (l*w1 == r*w2); //该子树成立 
    
}

int main(){
    int T,W;
    cin >> T;
    while(T--){
        if(solve(W)) cout << "YES" << endl; else cout << "NO" << endl;
        if(T) cout << endl;    //空行 
    }
    return 0;
}