839 - Not so Mobile

描述：这是物理上的一个杠杆问题，就是重物与力矩相乘，看一下两边是否相等，才开始我没有采用建树方法，后来又用建树的方法做了一下，不过很惭愧，建树时出现了问题，结果好几次wa……



代码一：

不建树代码



#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int flag;
int creat_recursion(int w1,int d1,int w2,int d2)
{
    int digit[5],w_left(w1),w_right(w2);
    if(!w1)
    {
        scanf("%d%d%d%d",&digit[0],&digit[1],&digit[2],&digit[3]);
        w_left=creat_recursion(digit[0],digit[1],digit[2],digit[3]);
    }
    if(!w2)
    {
        scanf("%d%d%d%d",&digit[0],&digit[1],&digit[2],&digit[3]);
        w_right=creat_recursion(digit[0],digit[1],digit[2],digit[3]);
    }
    if(w_left*d1!=w_right*d2)flag=1;
    return w_left+w_right;
}
int main()
{
    //freopen("a.txt","r",stdin);
    int n;
    scanf("%d",&n);
    int digit[5];
    while(n--)
    {
        scanf("%d%d%d%d",&digit[0],&digit[1],&digit[2],&digit[3]);
        flag=0;
        creat_recursion(digit[0],digit[1],digit[2],digit[3]);
        if(!flag)printf("YES\n");
        else printf("NO\n");
        if(n)printf("\n");
    }
    return 0;
}





代码二：

建树代码


#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
struct Node
{
    int w_left,d_left,w_right,d_right;
    Node *left,*right;
    Node ()
    {
        w_left=d_left=w_right=d_right=0;
        left=right=NULL;
    }
};
int flag,len;
Node *creat_tree(int p[100010][4])
{
    Node *tree=new Node;
    tree->w_left=p[len][0];
    tree->d_left=p[len][1];
    tree->w_right=p[len][2];
    tree->d_right=p[len][3];
    if(!tree->w_left)
    {
        len++;
        tree->left=creat_tree(p);
    }
    if(!tree->w_right)
    {
        len++;
        tree->right=creat_tree(p);
    }
    return tree;
}
int calculate_tree(Node *root)
{
    int w_left(root->w_left) , w_right( root->w_right ) ;
    if(root)
    {
        if( !root->w_left ) w_left = calculate_tree ( root->left);
        if( !root->w_right ) w_right = calculate_tree ( root->right);
    }
    if( w_left*root->d_left != w_right*root->d_right) flag=0;
    return w_left+w_right;
}
int main()
{
    //freopen("a.txt","r",stdin);
    int n,i,j;
    int digit[100010][4];
    scanf("%d",&n);
    while(n--)
    {
        i=len=0;
        j=2;
        memset(digit,0,sizeof(digit));
        while(scanf("%d%d%d%d",&digit[i][0],&digit[i][1],&digit[i][2],&digit[i][3])!=EOF)
        {
            j-=2;
            if(digit[i][0]==0)j+=2;
            if(digit[i][2]==0)j+=2;
            i++;
            if(!j)break;
        }
        Node *root=creat_tree(digit);
        flag=1;
        calculate_tree(root);
        if(flag)printf("YES\n");
        else printf("NO\n");
        if(n)printf("\n");
    }
    return 0;
}