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;
}