[CF_#660] C. Uncle Bogdan and Country Happiness

C. Uncle Bogdan and Country Happiness

1.分析：

本题可以用树上dfs预处理数据，再check的方式求解。
限制条件：
1.每个点的坏人数量是整数。
2.每个点的坏人数量大于子节点的坏人数量和+居民。（否则无法合理更新子节点）
3.每个点坏人的数量是正数。

2.算法描述

先树上dfs预处理出每个点会经过多少人，再判断（预处理）出每个点会经过多少坏人，最后check。

3.具体代码

#include <bits/stdc++.h>

using namespace std;

const int N = 2e5 + 10;

int h[N],e[N*2],ne[2*N],idx;//建树
int hh[N],ch[N];//hh和ch分别为每个点的读入信息（hh：因为会和h数组重名，所以叫hh；ch：常住的意思）
int bad[N],liu[N];//bad存储当前节点有几个坏人，liu表示当前节点一共会经过多少人（流量）
int sum[N];//表示当前点的儿子坏人数和，叶子节点为0
int n,m;
bool flag=true;
void init()
{
    memset(h,-1,sizeof h);
    idx=0;
    flag=true;
}
void add(int a,int b)
{
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}

void dfs(int u,int fa)
{
    liu[u]=ch[u];
    for(int i=h[u];~i;i=ne[i])
    {
        if(e[i]==fa)    continue;
        dfs(e[i],u);
    }
    for(int i=h[u];~i;i=ne[i])
    {
        if(e[i]==fa)    continue;
        liu[u]+=liu[e[i]];
    }
}

void dfs1(int u,int fa)
{
    sum[u]=0;
    for(int i=h[u];~i;i=ne[i])
    {
        if(e[i]==fa)    continue;
        dfs1(e[i],u);
    }
    for(int i=h[u];~i;i=ne[i])
    {
        if(e[i]==fa)    continue;
        sum[u]+=bad[e[i]];
    }
}

int main()
{
    int T ;
    scanf("%d",&T);
    while(T -- )
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&ch[i]);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&hh[i]);
        }
        for(int i=1;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            add(a,b),add(b,a);
        }
        dfs(1,0);//预处理出liu[]
        for(int i=1;i<=n;i++)//判断，并求出bad[]
        {
            if((liu[i]-hh[i])%2 == 0)   bad[i]=(liu[i]-hh[i])/2;
            else
            {
                flag=0;
                break;
            }
        }
        if(!flag)
        {
            printf("NO\n");
            continue;
        }
        dfs1(1,0);//预处理出sum[]
        for(int i=1;i<=n;i++)
        {
            if(bad[i]>ch[i]+sum[i] || bad[i]<0)
                flag=0;
        }
        if(flag)
        {
            printf("YES\n");
        }
        else
            printf("NO\n");
    }
    return 0;
}

4.总结

记录第一次打cf比赛。。
晚上思维比较混乱，白天才把代码实现出来。
因为漏掉了bad[]>0的限制条件卡了好久。
又因为没有建双向边一直卡第四个数据。