Chemistry in Berland CodeForces

题目

题意：

有n种化学物质，第i种物质现有bi千克，需要ai千克。有n-1种,编号为2-n的转换方式，每种都为(x,k)，第i行是编号为i+1的转换方式，编号为i的转换方式(xi,ki)表示ki千克的xi物质可以转换成1千克的i物质，1千克的i物质可以转换成1千克的xi物质。问是否可能通过转换得到足够的需要的物质。(1 ≤ x_j + 1 ≤ j)

重点：上面标红的条件。如果只保留ki千克的xi物质可以转换成1千克的i物质产生的一条有向边，表明xi物质连出的边一定是指向编号大于i的物质的（x[i+1]<=i,x[i]<=i-1,x[i]<i），而又恰好有n-1条边，也就是这种情况下这是一棵树。

方法：建树，dfs自底向上递推（我用的方法是只保留ki千克的xi物质可以转换成1千克的i物质产生的一条xi->i的有向边），如果某样物质不够就从父结点那儿转换，如果某样物质多了就把多的转换成父结点的物质。（如果父结点的物质不够，直接减就行，减成负数也没关系）

奇怪的地方：貌似这道题极限数据会爆longlong，然后直接在爆longlong的时候判为NO就行？

#include<cstdio>
#include<cstdlib>
#define inf 110000000000000000
typedef long long LL;
struct Edge
{
    LL to,dis,next;
}edge[100100];
LL n,num_edge;
LL first1[100100];
LL a[100100],b[100100];
double tem;
void dfs(LL x,LL fa,LL p)
{
    LL k=first1[x];
    while(k!=0)
    {
        dfs(edge[k].to,x,edge[k].dis);
        k=edge[k].next;
    }
//    if(a[x]<b[x])
//        b[fa]+=b[x]-a[x];
//    else if(a[x]>b[x])
//        b[fa]-=p*(a[x]-b[x]);
    if(a[x]<b[x])
        b[fa]+=b[x]-a[x];
    else if(a[x]>b[x])
    {
        tem=(double)(b[x]-a[x])*p;//为何要double？
        if(tem<-inf)
        {
            printf("NO");
            exit(0);
        }
        b[fa]-=p*(a[x]-b[x]);
        if(b[fa]<-inf)
        {
            printf("NO");
            exit(0);
        }
    }
}
int main()
{
    LL i,x,k;
    scanf("%lld",&n);
    for(i=1;i<=n;i++)
        scanf("%lld",&b[i]);
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(i=2;i<=n;i++)
    {
        scanf("%lld%lld",&x,&k);
        edge[++num_edge].to=i;
        edge[num_edge].dis=k;
        edge[num_edge].next=first1[x];
        first1[x]=num_edge;
    }
    k=first1[1];
    while(k!=0)
    {
        dfs(edge[k].to,1,edge[k].dis);
        k=edge[k].next;
    }
    if(b[1]<a[1])
        printf("NO");
    else
        printf("YES");
    return 0;
}