lightoj 1219 - Mafia 贪心

给定一颗有n个节点的树，每个节点都有一个人数，问将每个节点都有1个人最低的消耗是多少（每人走一步消耗1）。

输入比较扯，先输入节点u然后在输入这个点有多少人，再输入这个点与那些边相连了...

单向边就好了，毕竟是颗树，找到根应该就能遍历所有的点。

怎么移动人...从最底层考虑，叶子节点只能把多的人移动到父节点，不够也只能向父节点要。

那么完成了所有的叶子节点后，对于倒数第二层的节点，他们的孩子节点已经就位了...所以还是只能向父亲节点要人或者送人...

感觉有点像数塔的结构了...

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define N 10100
using namespace std;
struct edge
{
    int u,v,next;
}edge[N*2];
int head[N],a[N],in[N],tot,ans;
void init()
{
    memset(head,-1,sizeof(head));
    tot=0;
    memset(in,0,sizeof(in));
}
void add(int u,int v)
{
    edge[tot].v=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs(int u,int pre)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        dfs(v,u);
    }
    ans+=abs(a[u]-1);
    a[pre]+=a[u]-1;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int cas=1; cas<=t; cas++)
    {
        init();
        int n;
        scanf("%d",&n);
        int u,v,k;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&u);
            scanf("%d %d",&a[u],&k);
            while(k--)
            {
                scanf("%d",&v);
                in[v]++;
                add(u,v);
            }
        }
        int root=-1;
        for(int i=1;i<=n;i++)
        {
            if(in[i]==0)
                root=i;
        }
        ans=0;
        dfs(root,-1);
        printf("Case %d: %d\n",cas,ans);
    }
    return 0;
}