82分求助
查看原帖
82分求助
270791
WanderingTrader楼主2022/2/18 22:16

RT,TLE 2个点,不知道咋回事,求大佬帮忙康康

#include<bits/stdc++.h>
using namespace std;
#define id(x,y) ((x)*m+y-m)
#define in(x,y) id(x,y)
#define out(x,y) (id(x,y)+n*m)
const int N=25;
const int V=N*N;
const int E=1e3*V;
const int INF=0x3f3f3f3f;
struct edge{int v,c,nxt;} e[E];
int a[N][N],hd[V],cur[V],lv[V],tot=1,n,m,d,S,T,nV,cnt;
char s[N];
double dist(int x,int y,int X,int Y){return sqrt((X-x)*(X-x)+(Y-y)*(Y-y));}
void add(int u,int v,int c){e[++tot]={v,c,hd[u]};hd[u]=tot;}
bool bfs()
{
	memset(lv,0,sizeof(lv));
	queue<int> q;
	q.push(S);
	lv[S]=1;
	while(!q.empty())
	{
		int u=q.front();q.pop();
		for(int i=hd[u],v,c;i;i=e[i].nxt)
		{
			v=e[i].v; c=e[i].c;
			if(!c||lv[v]) continue;
			q.push(v);
			lv[v]=lv[u]+1;
		}
	}
	return lv[T];
}
int dfs(int u,int f)
{
	if(u==T) return f;
	int sum=0;
	for(int i=cur[u],v,c;i&&f;i=e[i].nxt)
	{
		cur[u]=i;
		v=e[i].v; c=e[i].c;
		if(!c||lv[v]!=lv[u]+1) continue;
		int delta=dfs(v,min(c,f));
		sum+=delta;
		f-=delta;
		e[i].c-=delta;
		e[i^1].c+=delta; 
	}
	return sum;
}
int Dinic()
{
	int maxflow=0;
	while(bfs())
	{
		for(int u=1;u<=nV;++u) cur[u]=hd[u];
		maxflow+=dfs(S,INF);
	}
	return maxflow;
}
int main()
{
	scanf("%d%d%d",&n,&m,&d);
	S=2*n*m+1; T=2*n*m+2; nV=T;
	for(int i=1;i<=n;++i) for(int j=1;j<=m;++j) scanf("%1d",&a[i][j]);
	for(int i=1,c;i<=n;++i) 
		for(int j=1;j<=m;++j)
		{
			c=a[i][j];
			if(c==0) continue; //没有石柱
			add(in(i,j),out(i,j),c);
			add(out(i,j),in(i,j),0); //入->出
			for(int x=1;x<=n;++x)
				for(int y=1;y<=m;++y) if((x!=i||y!=j)&&dist(i,j,x,y)<=d&&a[x][y]) //平面距离<=d且有石柱
					add(out(i,j),in(x,y),INF), add(in(x,y),out(i,j),0);
			if(i<=d||j<=d||n-i+1<=d||m-j+1<=d) 
				add(out(i,j),T,INF),add(T,out(i,j),0); //可以一步跳到界外
		}
	for(int i=1;i<=n;++i)
	{
		scanf("%s",s+1);
		for(int j=1;j<=m;++j) 
			if(s[j]=='L') 
				add(S,in(i,j),1),add(in(i,j),S,0),++cnt;
	}
	printf("%d\n",cnt-Dinic());
	return 0;
}

看上去带当前弧的dinic没有问题。(难道是define过多导致效率低下?)

2022/2/18 22:16
加载中...