KM迭代版 90分求助
查看原帖
KM迭代版 90分求助
270791
WanderingTrader楼主2022/6/28 22:42
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=505;
const LL INF=0x3f3f3f3f3f3f3f3f;
LL lx[N],ly[N],slack[N],w[N][N];
bool vy[N];
int a[N],b[N],l[N],r[N],pre[N],n,m,X,Y;
void bfs(int x)
{
	int u,v=0,yy=0;
	LL delta;
	memset(pre,0,sizeof(pre));
	memset(vy,0,sizeof(vy));
	memset(slack,0x3f,sizeof(slack));
	b[v]=x;
	do
	{
		u=b[v]; delta=INF; vy[v]=true;
		for(int y=1;y<=n;++y) if(!vy[y])
		{
			int d=lx[u]+ly[y]-w[u][y];
			if(slack[y]>d){slack[y]=d;pre[y]=v;}
			if(slack[y]<delta){delta=slack[y];yy=y;}
		}
		for(int i=0;i<=n;++i) 
		{
			if(vy[i]){lx[b[i]]-=delta;ly[i]+=delta;}
			else slack[i]-=delta;
		}
		v=yy;
	}while(b[v]);
	while(v) b[v]=b[pre[v]],v=pre[v];
}
void KM()
{
	for(int u=1;u<=n;++u) bfs(u); 
	LL ans=0;
	for(int i=1;i<=n;++i) if(b[i]) ans+=w[b[i]][i];
	printf("%lld\n",ans);
	for(int i=1;i<=Y;++i) a[b[i]]=w[b[i]][i]==0?0:i;
	for(int i=1;i<=Y;++i) printf("%d ",b[i]); puts("");
}
int main()
{
	freopen("P6577_11.in","r",stdin);
	freopen("P6577.out","w",stdout);
	memset(lx,~0x3f,sizeof(lx));   // 初始均为-INF
	scanf("%d%d",&n,&m);
	X=Y=n;
	for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) w[i][j]=-INF;
	for(int i=1,u,v;i<=m;++i)
	{
		scanf("%d%d",&u,&v);
		scanf("%lld",&w[u][v]);
		lx[u]=max(lx[u],w[u][v]);
	}
	KM();
	return 0;
}

rt,写了UOJ上的模板题,AC代码略作改动后交到这题,最后一个点WA,我的输出是-INF,不知道为啥。。。

码风……还算看得过去吧,求助大佬qwq

2022/6/28 22:42
加载中...