uva 839 Not so Mobile

链接:http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=10&page=show_problem&problem=780

题意是一个大天平有许多小天平组成,要保证每个天平的两端都是平衡的,即力臂*力矩相等。

从图片看起来就是一颗二叉树,用递归方式建树来计算每个子天平的左边重量和右边重量,并判断是否平衡。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<ctype.h>
#include<algorithm>
#include<string>
#define PI acos(-1.0)
#define maxn 160
#define INF 1<<25
#define MAX 0x7fffffff
typedef long long ll;
using namespace std;
struct Node
{
    int wl,dl,wr,dr;
    Node *lchild;
    Node *rchild;
};
Node *root;
int jud;
int Buildtree(Node *&node,int a,int b,int c,int d)
{
    node=(Node*)malloc(sizeof(Node));
    int q,w,e,r;
    node->dl=b;
    node->dr=d;
    if(a>0)
        node->wl=a;
    else
    {
        scanf("%d%d%d%d",&q,&w,&e,&r);
        Buildtree(node->lchild,q,w,e,r);
        node->wl=node->lchild->wl+node->lchild->wr;
    }
    if(c>0)
        node->wr=c;
    else
    {
        scanf("%d%d%d%d",&q,&w,&e,&r);
        Buildtree(node->rchild,q,w,e,r);
            node->wr=node->rchild->wl+node->rchild->wr;
    }
}
int dfs(Node *&node)
{
    if(jud==1)
        return 0;
    if(node->wl*node->dl!=node->wr*node->dr)
    {
        jud=1;
        return 0;
    }
    else
    {
        if(node->lchild!=NULL)
            dfs(node->lchild);
        if(node->rchild!=NULL)
            dfs(node->rchild);
    }
    return 0;
}
int main()
{
    int tot;
    scanf("%d",&tot);
    for(int i=0; i<tot; i++)
    {
        jud=0;
        int a,b,c,d;
        scanf("%d%d%d%d",&a,&b,&c,&d);
        Buildtree(root,a,b,c,d);
        dfs(root);
        if(jud==0)
            printf("YES\n");
        else printf("NO\n");
        if(i<tot-1)
            printf("\n");
    }
}


uva 839 Not so Mobile

上一篇:eclipse的maven项目总是在关联源码


下一篇:Knight Moves