紫书章六例题九 天平 UVA 839（更新二叉树的节点上的值）

题意：输入一个树状天平，看是否都符合力矩相等。递归输入每个节点（天平）。第一种是用指针来建树，然后逐步向上更新判断

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

struct node
{
    int v,w1,d1,w2,d2;
    node *l,*r;
    node(int a=0,node *left=NULL,node *right=NULL)
    {
        v=a,l=left,r=right;
    }
};
node *root;
int flag=0;
node* buildtree(node *t)
{
    int w1,d1,w2,d2;
    scanf("%d %d %d %d",&w1,&d1,&w2,&d2);
    t->d1=d1,t->d2=d2,t->w1=w1,t->w2=w2;
    if(!w1) {
        if(t->l==NULL){t->l=new node();}
        buildtree(t->l);
    }
    if(!w2){
        if(t->r==NULL) {t->r=new node();}
        buildtree(t->r);
    }
    if(t->l!=NULL)//更新上面节点的W
    t->w1+=(t->l)->w1+(t->l)->w2;
    if(t->r!=NULL)
    t->w2+=(t->r)->w1+(t->r)->w2;
    if(t->w1*t->d1!=t->w2*t->d2) flag=1;
}
int main()
{
    int T,case1=0;
    scanf("%d",&T);
    while(T--)
    {
        if(case1) printf("\n");
        flag=0;
        root =new node();
        buildtree(root);
        if(flag==1) printf("NO\n");
        else printf("YES\n");
        case1=1;
    }
    return 0;
}

第二种是用引用的方法，代码更精简

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <sstream>
#include <cmath>

using namespace std;
int flag=0;
void solve(int &w)
{
    int w1,d1,w2,d2;
    cin>>w1>>d1>>w2>>d2;
    if(!w1) solve(w1);
    if(!w2) solve(w2);
    w=w1+w2;
    if(w1*d1!=w2*d2) flag=1;
}
int main()
{
    int T,w;
    scanf("%d",&T);
    while(T--)
    {
        flag=0;
        solve(w);
        if(!flag) printf("YES\n");else printf("NO\n");
        if(T) printf("\n");
    }
    return 0;
}