BZOJ1066(SCOI2007)[蜥蜴]--最大流(Dinic)

【链接】
bzoj1066

【解题报告】

拆点加超级源和超级汇建图，刷Dinic。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=805,maxm=26805,INF=((1<<30)-1)*2+1;
int n,m,D,tot,sum,st,gl,cur[maxn],lnk[maxn],que[maxn],dst[maxn],w[maxm],son[maxm],nxt[maxm];
bool vis[maxn];
inline char nc()
{
    static char buf[100000],*l,*r;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int Read()
{
    int res=0; char ch=nc();
    while (ch<'0'||ch>'9') ch=nc();
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
    return res;
}
char Readc()
{
    char ch=nc();
    while ((ch<'0'||ch>'3')&&ch!='L'&&ch!='.') ch=nc();
    return ch;
}
int Getid(int x,int y,int z) {return z*n*m+(x-1)*m+y;}
int Sqr(int x) {return x*x;}
void Add(int x,int y,int z)
{
    w[++tot]=z; son[tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
    w[++tot]=0; son[tot]=x; nxt[tot]=lnk[y]; lnk[y]=tot;
}
int Bfs(int s,int t)
{
    memset(vis,0,sizeof(vis));
    memset(dst,0,sizeof(dst));
    int hed=0,til=1; que[1]=s; vis[s]=1;
    while (hed!=til)
    {
        int x=que[++hed];
        for (int j=lnk[x]; ~j; j=nxt[j])
         if (!vis[son[j]]&&w[j])
          que[++til]=son[j],vis[son[j]]=1,dst[son[j]]=dst[x]+1;
    }
    return vis[t];
}
int Dfs(int x,int t,int MIN)
{
    if (x==t||!MIN) return MIN; int Flow=0;
    for (int &j=cur[x]; ~j; j=nxt[j])
     if (dst[x]+1==dst[son[j]])
      {
        int now=Dfs(son[j],t,min(MIN,w[j]));
        if (now)
         {
            w[j]-=now; w[j^1]+=now; Flow+=now; MIN-=now;
            if (!MIN) break;
        }
      }
    return Flow;
}
int MAXflow(int s,int t)
{
    int Flow=0;
    while (Bfs(s,t))
    {
        memcpy(cur,lnk,sizeof(cur));
        Flow+=Dfs(s,t,INF);
    }
    return Flow;
}
int main()
{
    freopen("1066.in","r",stdin);
    freopen("1066.out","w",stdout);
    n=Read(); m=Read(); D=Read(); tot=1; sum=0; st=0; gl=2*n*m+1;
    memset(lnk,255,sizeof(lnk));
    for (int i=1; i<=n; i++)
     for (int j=1; j<=m; j++)
      {
        int x=Readc()-48;
        if (x) Add(Getid(i,j,0),Getid(i,j,1),x);
      }
    for (int i=1; i<=n; i++)
     for (int j=1; j<=m; j++)
      if (Readc()=='L') sum++,Add(st,Getid(i,j,0),1);
    for (int i=1; i<=n; i++)
     for (int j=1; j<=m; j++)
      for (int x=max(1,i-D); x<=min(n,i+D); x++)
       for (int y=max(1,j-D); y<=min(m,j+D); y++)
        if ((i!=x||j!=y)&&Sqr(x-i)+Sqr(j-y)<=Sqr(D))
         Add(Getid(i,j,1),Getid(x,y,0),INF);
    for (int i=1; i<=n; i++)
     for (int j=1; j<=m; j++)
      if (min(min(i,j),min(n-i+1,m-j+1))<=D)
       Add(Getid(i,j,1),gl,INF);
    printf("%d\n",sum-MAXflow(st,gl));
    return 0;
}