Leapin' Lizards（经典建图，最大流） Leapin' Lizards

  1 #include<iostream>
  2 #include<cstring>
  3 #include<string>
  4 #include<cmath>
  5 #include<cstdio>
  6 #include<algorithm>
  7 #include<queue>
  8 #include<vector>
  9 #include<set>
 10 #define maxn 200005
 11 #define MAXN 200005
 12 #define mem(a,b) memset(a,b,sizeof(a))
 13 const int N=200005;
 14 const int M=200005;
 15 const int INF=0x3f3f3f3f;
 16 using namespace std;
 17 int n;
 18 struct Edge{
 19     int v,next;
 20     int cap,flow;
 21 }edge[MAXN*20];//注意这里要开的够大。。不然WA在这里真的想骂人。。问题是还不报RE。。
 22 int cur[MAXN],pre[MAXN],gap[MAXN],path[MAXN],dep[MAXN];
 23 int cnt=0;//实际存储总边数
 24 void isap_init()
 25 {
 26     cnt=0;
 27     memset(pre,-1,sizeof(pre));
 28 }
 29 void isap_add(int u,int v,int w)//加边
 30 {
 31     edge[cnt].v=v;
 32     edge[cnt].cap=w;
 33     edge[cnt].flow=0;
 34     edge[cnt].next=pre[u];
 35     pre[u]=cnt++;
 36 }
 37 void add(int u,int v,int w){
 38     isap_add(u,v,w);
 39     isap_add(v,u,0);
 40 }
 41 bool bfs(int s,int t)//其实这个bfs可以融合到下面的迭代里，但是好像是时间要长
 42 {
 43     memset(dep,-1,sizeof(dep));
 44     memset(gap,0,sizeof(gap));
 45     gap[0]=1;
 46     dep[t]=0;
 47     queue<int>q;
 48     while(!q.empty())
 49     q.pop();
 50     q.push(t);//从汇点开始反向建层次图
 51     while(!q.empty())
 52     {
 53         int u=q.front();
 54         q.pop();
 55         for(int i=pre[u];i!=-1;i=edge[i].next)
 56         {
 57             int v=edge[i].v;
 58             if(dep[v]==-1&&edge[i^1].cap>edge[i^1].flow)//注意是从汇点反向bfs，但应该判断正向弧的余量
 59             {
 60                 dep[v]=dep[u]+1;
 61                 gap[dep[v]]++;
 62                 q.push(v);
 63                 //if(v==sp)//感觉这两句优化加了一般没错，但是有的题可能会错，所以还是注释出来，到时候视情况而定
 64                 //break;
 65             }
 66         }
 67     }
 68     return dep[s]!=-1;
 69 }
 70 int isap(int s,int t)
 71 {
 72     if(!bfs(s,t))
 73     return 0;
 74     memcpy(cur,pre,sizeof(pre));
 75     //for(int i=1;i<=n;i++)
 76     //cout<<"cur "<<cur[i]<<endl;
 77     int u=s;
 78     path[u]=-1;
 79     int ans=0;
 80     while(dep[s]<n)//迭代寻找增广路,n为节点数
 81     {
 82         if(u==t)
 83         {
 84             int f=INF;
 85             for(int i=path[u];i!=-1;i=path[edge[i^1].v])//修改找到的增广路
 86                 f=min(f,edge[i].cap-edge[i].flow);
 87             for(int i=path[u];i!=-1;i=path[edge[i^1].v])
 88             {
 89                 edge[i].flow+=f;
 90                 edge[i^1].flow-=f;
 91             }
 92             ans+=f;
 93             u=s;
 94             continue;
 95         }
 96         bool flag=false;
 97         int v;
 98         for(int i=cur[u];i!=-1;i=edge[i].next)
 99         {
100             v=edge[i].v;
101             if(dep[v]+1==dep[u]&&edge[i].cap-edge[i].flow)
102             {
103                 cur[u]=path[v]=i;//当前弧优化
104                 flag=true;
105                 break;
106             }
107         }
108         if(flag)
109         {
110             u=v;
111             continue;
112         }
113         int x=n;
114         if(!(--gap[dep[u]]))return ans;//gap优化
115         for(int i=pre[u];i!=-1;i=edge[i].next)
116         {
117             if(edge[i].cap-edge[i].flow&&dep[edge[i].v]<x)
118             {
119                 x=dep[edge[i].v];
120                 cur[u]=i;//常数优化
121             }
122         }
123         dep[u]=x+1;
124         gap[dep[u]]++;
125         if(u!=s)//当前点没有增广路则后退一个点
126         u=edge[path[u]^1].v;
127      }
128      return ans;
129 }
130 
131 string mp[505];
132 string book[505];
133 int dir[4][2]={0,1,1,0,0,-1,-1,0};
134 int main(){
135     std::ios::sync_with_stdio(false);
136     int m,s,t;
137     int f,d;
138     int T;
139     cin>>T;
140     for(int co=1;co<=T;co++){
141         cin>>n>>d;
142         for(int i=0;i<n;i++) cin>>book[i];
143         for(int i=0;i<n;i++) cin>>mp[i];
144         isap_init();
145         m=mp[0].length();
146         int sum=0;
147         s=n*m*2,t=n*m*2+1;
148         for(int i=0;i<n;i++){
149             for(int j=0;j<m;j++){
150                 if(mp[i][j]=='L'){
151                     add(s,i*m+j,1);
152                     sum++;
153                 }
154             }
155         }
156         for(int i=0;i<n;i++){
157             for(int j=0;j<m;j++){
158                 if(book[i][j]!='0'){
159                     if(i<d||j<d||i+d>=n||j+d>=m){
160                         add(m*n+i*m+j,t,book[i][j]-'0');
161                     }
162                     add(i*m+j,n*m+i*m+j,book[i][j]-'0');
163                     for(int ii=-d;ii<=d;ii++){
164                         for(int jj=-d;jj<=d;jj++){
165                             int xx=i+ii,yy=j+jj;
166                             if(xx>=0&&xx<n&&yy>=0&&yy<m){
167                                 if(book[xx][yy]!='0'&&(ii||jj)){
168                                     if(abs(ii)+abs(jj)<=d){
169                                         add(n*m+i*m+j,xx*m+yy,INF);
170                                     }
171                                 }
172                             }
173                         }
174                     }
175                 }
176             }
177         }
178         n=n*m*2+2;
179         int ans=isap(s,t);
180         ans=sum-ans;
181         cout<<"Case #"<<co<<": ";
182         if(!ans) cout<<"no lizard was left behind."<<endl;
183         else if(ans==1) cout<<"1 lizard was left behind."<<endl;
184         else cout<<ans<<" lizards were left behind."<<endl;
185     }
186 }