1412: [ZJOI2009]狼和羊的故事

题目链接

题目大意：矩阵中有0,1,2，将1和2隔开，0随意，求最小代价

题解：最小割模型比较显然……直接暴力建图，每个点都向四周连边就能过啦

我的收获：最小割第一题get

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
const int M=10005;
const int dx[]={0,1,-1,0,0};
const int dy[]={0,0,0,1,-1};

bool Exit;
int n,m,t,st,ed,tot;
int mp[105][105],id[105][105];
int num[M],d[M],head[M],last[M]; 

struct edge{
    int to,nex,c;
}e[M*20];

void add(int i,int j,int w){e[t].to=j,e[t].c=w,e[t].nex=head[i],last[i]=head[i]=t++;}
void insert(int x,int y,int z){add(x,y,z),add(y,x,0);};

int dfs(int x,int in)
{
    if(x==ed) return in;
    int ans=0,t;
    for(int i=last[x];i!=-1;last[x]=i=e[i].nex)
    {
        int v=e[i].to;
        if(e[i].c&&d[v]==d[x]-1){
            t=dfs(v,min(in-ans,e[i].c));
            ans+=t;
            e[i].c-=t; e[i^1].c+=t;
            if(Exit||ans==in) return ans;
        }
    }
    if(--num[d[x]]==0) Exit=1;
    d[x]++,num[d[x]]++,last[x]=head[x];
    return ans;
}

void build()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            if(mp[i][j]==1) insert(st,id[i][j],INF);
            if(mp[i][j]==2) insert(id[i][j],ed,INF);
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            for(int k=1;k<=4;k++){
                int fx=i+dx[k];
                int fy=j+dy[k];
                if(!id[fx][fy]) continue;
                insert(id[i][j],id[fx][fy],1);
            }
}

void work()
{
    Exit=0;
    int flow=0;
    while(!Exit) flow+=dfs(st,INF);
    printf("%d\n",flow);  

}

void init()
{
    memset(head,-1,sizeof(head));
    memset(last,-1,sizeof(last));
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&mp[i][j]),id[i][j]=++tot;
    st=0,ed=tot+1;num[0]=tot+2;
    build();
}

int main()
{
    init();
    work();
    return 0;
}