[NOI2009][BZOJ1565] 植物大战僵尸

1565: [NOI2009]植物大战僵尸

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 1654  Solved: 769
[Submit][Status][Discuss]

Description

Input

Output

仅包含一个整数，表示可以获得的最大能源收入。注意，你也可以选择不进行任何攻击，这样能源收入为0。

Sample Input

3 2
10 0
20 0
-10 0
-5 1 0 0
100 1 2 1
100 0

Sample Output

25

HINT

在样例中, 植物P1,1可以攻击位置(0,0), P2, 0可以攻击位置(2,1)。 
一个方案为，首先进攻P1,1, P0,1，此时可以攻击P0,0 。共得到能源收益为(-5)+20+10 = 25。注意, 位置(2,1)被植物P2,0保护，所以无法攻击第2行中的任何植物。 
【大致数据规模】
约20%的数据满足1 ≤ N, M ≤ 5；
约40%的数据满足1 ≤ N, M ≤ 10；
约100%的数据满足1 ≤ N ≤ 20，1 ≤ M ≤ 30，-10000 ≤ Score ≤ 10000 。

Source

首先拓扑排序找环，从环上的每个点开始dfs找出所有保护的点，把这些点全都去掉。最后重建图，就是最大权闭合图，跑最大流就好了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<queue>
#define INF 100000007
using namespace std;
int sumx,ans,edge,n,m,sum,x,y,num,tot;
int w[666],dis[666],q[666],head[666],head0[666],list[500000],list0[500000],next[500000],next0[500000],key[500000];
int ind[666],topo[666];
bool v[666];
void insert(int x,int y)
{
    next[++edge]=head[x];
    head[x]=edge;
    list[edge]=y;
}
void insert0(int x,int y,int z)
{
    next0[++edge]=head0[x];
    head0[x]=edge;
    list0[edge]=y;
    key[edge]=z;
}
int find()
{
    for (int i=1;i<=n*m;i++) if (!ind[i]) return i;
    return -1;
}
void toposort()
{
    for (int i=1;i<=n*m;i++)
    {
        int k=find();
        if (k==-1) break;
        topo[++tot]=k;
        ind[k]=-1;
        for (int j=head[k];j;j=next[j]) ind[list[j]]--;
    }
}
void dfs(int x)
{
    for (int i=head[x];i;i=next[i]) 
        if (v[list[i]])
        {
            v[list[i]]=0;
            dfs(list[i]);
        }
}
void rebuild()
{
    edge=1;
    for (int i=1;i<=n*m;i++) 
    {
        if (!v[i]) continue;
        if (w[i]>0)
        {
            insert0(0,i,w[i]);
            insert0(i,0,0);
        }
        if (w[i]<0)
        {
            insert0(i,n*m+1,-w[i]);
            insert0(n*m+1,i,0);
        }
        for (int j=head[i];j;j=next[j])
            if (v[list[j]]) 
            {
                insert0(list[j],i,INF);
                insert0(i,list[j],0);
            }
    }
}
bool BFS()
{
    memset(dis,0xff,sizeof(dis));
    int x,t=0,w=1;
    q[1]=0; dis[0]=1;
    while (t<w)
    {
        x=q[++t];
        for (int i=head0[x];i;i=next0[i])
            if (key[i]&&dis[list0[i]]==-1)
            {
                dis[list0[i]]=dis[x]+1;
                q[++w]=list0[i];
            }
    }
    return dis[n*m+1]!=-1;
}
int find(int x,int flow)
{
    if (x==n*m+1) return flow;
    int used=0,a;
    for (int i=head0[x];i;i=next0[i])
        if (key[i]&&dis[list0[i]]==dis[x]+1)
        {
                                        a=flow-used;
                                        a=find(list0[i],min(key[i],a));
                                        key[i]-=a;
                                        key[i^1]+=a;
                                        used+=a;
                                        if (used==flow) return flow;
        }
    if (!used) dis[x]=-1;
    return used;
}
int main()
{
    memset(v,0,sizeof(v));
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n*m;i++)
    {
        scanf("%d%d",&w[i],&sum);
        for (int j=1;j<=sum;j++)
        {
            scanf("%d%d",&x,&y);
            num=x*m+y+1;
            insert(i,num);
            ind[num]++;
        }
    }
    for (int i=0;i<n;i++)
        for (int j=0;j<m-1;j++)
        {
            num=i*m+j+1;
            insert(num+1,num);
            ind[num]++;
        }
    toposort();
    for (int i=1;i<=tot;i++) v[topo[i]]=1;
    for (int i=1;i<=n*m;i++) 
        if (!v[i]) dfs(i);
    for (int i=1;i<=n*m;i++) 
        if (w[i]>0&&v[i]) sumx+=w[i];
    rebuild();
    ans=0;
    while (BFS()) ans+=find(0,INF); 
    printf("%d",sumx-ans);
    return 0;
}