【BZOJ1565】 植物大战僵尸

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

 

---

Solution

　　按照依赖关系建立有向图：若清除$a$前必须清除$b$，则连边$a$至$b$。这些依赖关系包括同行植物左边对右边的依赖，以及被保护者对保护者的依赖。

　　首先不能考虑开挂集团，也就是依赖关系成强联通分量的植物，也包括能通过依赖关系走到强联通分量的植物——它们都无敌，需要排除掉。这可以用tarjan加上反向边的深搜预处理。

　　

　　一个植物能够下手，当且仅当其出度为0.

　　每一株植物又是带正负权值的，那么明显是要在依赖图中，求一个最大权闭合子图。也就是要有植物起手（迎合闭合的性质，出边都在闭合子图内，即依赖的植物都要一起干掉，都应该在闭合子图的范围内），且干掉的植物权值和最大。

　　按照最大权闭合子图的建立方法：源点向所有正权点连容量为其权值的边，所有负权点向汇点连容量为其权值绝对值的边；所有节点按依赖关系连边，容量为$+\infty$。

　　答案是所有正权点点权值和减去最大流。能将负权转化为正权来跑网络流的原因，拆拆括号就可以发现，这是很巧妙的。

---

#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
const int N=,INF=;
int n,m,a[N],h[N],hr[N],tot;
int dfn[N],low[N],tmcnt,is[N],st[N],top,ins[N],vis[N];
int S,T,dis[N],cur[N];
int sum;
queue<int> q;
vector<int> list[N];
struct Edge{int v,f,next;}g[];
inline int min(int x,int y){return x<y?x:y;}
inline int id(int x,int y){return (x-)*m+y;}
inline void addEdge(int u,int v){
    g[++tot].v=v; g[tot].next=h[u]; h[u]=tot;
    g[++tot].v=u; g[tot].next=hr[v]; hr[v]=tot;
}
inline void addEdge(int u,int v,int f){
    g[++tot].v=v; g[tot].f=f; g[tot].next=h[u]; h[u]=tot;
    g[++tot].v=u; g[tot].f=; g[tot].next=h[v]; h[v]=tot;
}
bool bfs(){
    while(!q.empty()) q.pop();
    q.push(S);
    for(int i=;i<=T;i++) dis[i]=-;
    dis[S]=;
    while(!q.empty()){
        int u=q.front(); q.pop();
        for(int i=h[u],v;i;i=g[i].next)
            if(g[i].f&&dis[v=g[i].v]==-){
                dis[v]=dis[u]+;
                if(v==T) return true;
                q.push(v);
            }
    }
    return dis[T]!=-;
}
int dfs(int u,int delta){
    if(u==T) return delta;
    int ret=,get;
    for(int i=cur[u],v;i&&delta;i=g[i].next)
        if(g[i].f&&dis[v=g[i].v]==dis[u]+){
            get=dfs(v,min(delta,g[i].f));
            g[i].f-=get; g[i^].f+=get;
            if(g[i].f) cur[u]=i;
            delta-=get; ret+=get;
        }
    if(!ret) dis[u]=-;
    return ret;
}
int dinic(){
    int ret=;
    while(bfs()){
        for(int i=;i<=T;i++) cur[i]=h[i];
        ret+=dfs(S,INF);
    }
    return ret;
}
void tarjan(int u,int fa){
    dfn[u]=low[u]=++tmcnt;
    st[++top]=u; ins[u]=;
    for(int i=h[u],v;i;i=g[i].next){
        v=g[i].v;
        if(!dfn[v]){
            tarjan(v,u);
            low[u]=min(low[u],low[v]);
        }
        else if(ins[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u]){
        if(st[top]==u){
            ins[u]=;
            top--;
            return;
        }
        int x;
        do{
            x=st[top--];
            ins[x]=;
            is[x]=;
        }while(x!=u);
    }
}
void clear(int u){
    vis[u]=; is[u]=;
    for(int i=hr[u],v;i;i=g[i].next)
        if(!vis[v=g[i].v])
            clear(v);
}
int main(){
    freopen("input.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++)
        for(int j=;j<=m;j++){
            int k=id(i,j),l,x,y;
            scanf("%d%d",&a[k],&l);
            while(l--){
                scanf("%d%d",&x,&y);
                addEdge(id(x+,y+),k);
                list[id(x+,y+)].push_back(k);
            }
        }
    for(int i=;i<=n;i++)
        for(int j=;j<m;j++)
            addEdge(id(i,j),id(i,j+));
    for(int i=,up=n*m;i<=up;i++)
        if(!dfn[i])
            tarjan(i,);
    for(int i=,up=n*m;i<=up;i++)
        if(!vis[i]&&is[i])
            clear(i);
    for(int i=,up=n*m;i<=up;i++) h[i]=;
    tot=;
    S=n*m+; T=n*m+;
    for(int i=;i<=n;i++)
        for(int j=;j<=m;j++){
            int x=id(i,j);
            if(j<m&&!is[x]&&!is[id(i,j+)])
                addEdge(x,id(i,j+),INF);
            if(is[x]) continue;
            if(a[x]>=){
                addEdge(S,x,a[x]);
                sum+=a[x];
            }
            else addEdge(x,T,-a[x]);
            int sz=list[x].size();
            for(int k=;k<sz;k++)
                addEdge(x,list[x][k],INF);
        }
    int maxflow=dinic();
    printf("%d\n",sum-maxflow);
    return ;
}

奇妙代码