按照第一篇题解的思路,敲出了如下代码。
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define lll __int128
#define db double
#define ld long double
#define eps 1e-8
#define fir first
#define sec second
#define pb push_back
#define ppc __builtin_popcount
using namespace std;
const int maxn=16400,inf=1e9;
int N,a[130][130],c[130][130];
inline int g(int i,int j){return (i-1)*N+j;}
namespace FLOW{
struct edge{int v,w,prev;}E[maxn*10];
int S,T,head[maxn<<1],cnt;
inline void add(int u,int v,int w){E[++cnt]={v,w,head[u]};head[u]=cnt;}
inline void add_edge(int u,int v,int w){add(u,v,w),add(v,u,0);}
int now[maxn<<1],d[maxn<<1];
bool bfs(){
for(int i=0;i<=T;i++) d[i]=0,now[i]=head[i];
queue<int> q;
q.push(S);
d[S]=1;
while(!q.empty()){
int u=q.front();q.pop();
for(int i=head[u],v=E[i].v;i;v=E[i=E[i].prev].v)
if(E[i].w&&!d[v])
d[v]=d[u]+1,q.push(v);
}
return d[T];
}
inline int dfs(int u,int flow){
if(u==T||!flow) return flow;
int val=0;
for(int i=now[u],v=E[i].v;i;v=E[i=E[i].prev].v){
now[u]=i;
if(!E[i].w||d[v]!=d[u]+1) continue;
int t=dfs(v,min(flow,E[i].w));
val+=t,flow-=t;
E[i&1?i+1:i-1].w+=t,E[i].w-=t;
if(!flow) break;
}
if(!val) d[u]=0;
return val;
}
int dinic(){
int ans=0,t;
while(bfs())
while((t=dfs(S,inf))>0)
ans+=t;
return ans;
}
}//namespace FLOW
using FLOW::add_edge;
using FLOW::S;
using FLOW::T;
mt19937 RND(time(0));
inline int rnd(int l,int r){return RND()%(r-l+1)+l;}
void gen(){
N=128;
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
a[i][j]=(i+j-2)%N+1;
for(int i=1;i<=N;i++){
int x=rnd(1,N),y=rnd(1,N);
for(int j=1;j<=N;j++) swap(a[x][j],a[y][j]);
x=rnd(1,N),y=rnd(1,N);
for(int j=1;j<=N;j++) swap(a[j][x],a[j][y]);
}
cerr<<N<<"\n";
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
c[i][j]=rnd(0,1);
for(int i=1;i<=N;i++,cerr<<"\n")
for(int j=1;j<=N;j++)
cerr<<a[i][j]<<" ";
for(int i=1;i<=N;i++,cerr<<"\n")
for(int j=1;j<=N;j++)
cerr<<c[i][j]<<" ";
}
int main(){
freopen("log.txt","w",stderr);
// gen();
cin>>N;
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
cin>>a[i][j];
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
cin>>c[i][j];
S=0,T=2*N*N+1;
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
int u=g(i,j);
add_edge(u,u+N*N,1-c[i][j]+2*N);
if(a[i][j]==1) add_edge(S,u,inf);
if(a[i][j]==N) add_edge(u+N*N,T,inf);
for(int k=1;k<=N;k++){
if(a[i][k]==a[i][j]+1) add_edge(u+N*N,g(i,k),inf);
if(a[k][j]==a[i][j]+1) add_edge(u+N*N,g(k,j),inf);
}
}
}
int flow=FLOW::dinic();
// cerr<<"flow = "<<flow<<"\n";
printf("%d\n",N-flow+2*N*N);
return 0;
}
会 TLE on 12。
但是如果将 108 至 111 行换成
for(int k=1;k<=N;k++)
if(a[i][k]==a[i][j]+1)
add_edge(u+N*N,g(i,k),inf);
for(int k=1;k<=N;k++)
if(a[k][j]==a[i][j]+1)
add_edge(u+N*N,g(k,j),inf);
就过了,而且跑的飞快。
这是为什么呢?