BZOJ1066蜥蜴

1066: [SCOI2007]蜥蜴
Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 2843 Solved: 1413
Description
在一个r行c列的网格地图中有一些高度不同的石柱，一些石柱上站着一些蜥蜴，你的任务是让尽量多的蜥蜴逃到边界外。 每行每列中相邻石柱的距离为1，蜥蜴的跳跃距离是d，即蜥蜴可以跳到平面距离不超过d的任何一个石柱上。石柱都不稳定，每次当蜥蜴跳跃时，所离开的石柱高度减1（如果仍然落在地图内部，则到达的石柱高度不变），如果该石柱原来高度为1，则蜥蜴离开后消失。以后其他蜥蜴不能落脚。任何时刻不能有两只蜥蜴在同一个石柱上。
Input
输入第一行为三个整数r，c，d，即地图的规模与最大跳跃距离。以下r行为石竹的初始状态，0表示没有石柱，1~3表示石柱的初始高度。以下r行为蜥蜴位置，“L”表示蜥蜴，“.”表示没有蜥蜴。
Output
输出仅一行，包含一个整数，即无法逃离的蜥蜴总数的最小值。
Sample Input
5 8 2
00000000
02000000
00321100
02000000
00000000
……..
……..
..LLLL..
……..
……..
Sample Output
1
HINT
100%的数据满足：1<=r, c<=20, 1<=d<=4
Source
Pku 2711 Leapin’ Lizards
最大流+拆点
将石柱一分为二，连线容量为石柱高度。。
源点与地图中的所有蜥蜴相连，容量为1
汇点与地图中所有蜥蜴能跳出的点相连，容量为inf
如果任意两个石柱间距小于d，就把一个石柱的底部和另一个石柱的顶部相连，容量为inf。。
最后用蜥蜴数量减去最大流就是结果。。
附上本蒟蒻的代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define inf 0x7fffffff
#define T 801
int r,c,d,cnt=1,ans=0,sum,map[21][21],mark[21][21],q[802],h[802],dis[802],head,tail;
struct kx
{
    int to,next,v;
}edge[500001];

int read()
{
    int w=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') f=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*f;
}

void add(int u,int v,int w)
{
    cnt++,edge[cnt].next=h[u],h[u]=cnt,edge[cnt].to=v,edge[cnt].v=w;
    cnt++,edge[cnt].next=h[v],h[v]=cnt,edge[cnt].to=u,edge[cnt].v=0;
}

bool bfs()
{
    int j,p;
    memset(dis,-1,sizeof(dis));
    head=0,tail=1,dis[0]=q[0]=0;
    while (head<tail)
      {
        head++,j=q[head],p=h[j];
        while (p)
          {
            if (dis[edge[p].to]<0 && edge[p].v>0)
              dis[edge[p].to]=dis[j]+1,tail++,q[tail]=edge[p].to;
            p=edge[p].next;
          }
      }
    if (dis[T]>0) return true;
    else return false;
}

int dfs(int x,int f)
{
    int w,used=0,i=h[x];
    if (x==T) return f;
    while (i)
      {
        if (dis[edge[i].to]==dis[x]+1 && edge[i].v)
          {
            w=f-used,w=dfs(edge[i].to,min(w,edge[i].v)),edge[i].v-=w,edge[i^1].v+=w,used+=w;
            if (used==f) return f;
          }
        i=edge[i].next;
      }
    if (!used) dis[x]=-1;
    return used;
}

bool judge(int x1,int y1,int x2,int y2)
{
    if (((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2))<=(d*d) && map[x1][y1] && map[x2][y2]) return true;
    return false;
}

void build()
{
    for (int x1=1;x1<=r;x1++)
      for (int y1=1;y1<=c;y1++)
        for (int x2=x1-d;x2<=x1+d;x2++)
          for (int y2=y1-d;y2<=y1+d;y2++)
            if (judge(x1,y1,x2,y2) && (x1!=x2 || y1!=y2))
              add(mark[x1][y1]+400,mark[x2][y2],inf);
    for (int i=1;i<=r;i++)
      for (int j=1;j<=c;j++)
        if (map[i][j]) add(mark[i][j],mark[i][j]+400,map[i][j]);
}

int main()
{
    int i,j,tot=0;
    r=read(),c=read(),d=read();
    char s[21];
    for (i=1;i<=r;i++)
      {
        scanf("%s",&s);
        for (j=1;j<=c;j++)
          map[i][j]=s[j-1]-'0';
      }
    for (i=1;i<=r;i++)
      for (j=1;j<=c;j++)
        tot++,mark[i][j]=tot;
    for (i=1;i<=r;i++)
      {
        scanf("%s",&s);
        for (j=1;j<=c;j++)
          if (s[j-1]=='L')
            add(0,mark[i][j],1),ans++;
      }
    for (i=1;i<=d;i++)
      for (j=d+1;j<=r-d;j++)
        add(mark[j][i]+400,T,inf),add(mark[j][c-i+1]+400,T,inf);
    for (i=1;i<=d;i++)
      for (j=1;j<=c;j++)
        add(mark[i][j]+400,T,inf),add(mark[r-i+1][j]+400,T,inf);
    build();
    while (bfs())
      while (sum=dfs(0,inf))
        ans-=sum;
    printf("%d",ans);
    return 0;
}