UOJ#80. 二分图最大权匹配(BFS)

从前一个和谐的班级，有 nl 个是男生，有 nr 个是女生。编号分别为 1,…,nl 和 1,…,nr。

有若干个这样的条件：第 v 个男生和第 u 个女生愿意结为配偶，且结为配偶后幸福程度为 ww。

请问这个班级里幸福程度之和最大是多少？

输入格式

第一行三个正整数，nl,nr,m。

接下来 mm 行，每行三个整数 v,u,w 表示第 v 个男生和第 u 个女生愿意结为配偶，且幸福程度为 w。保证 1≤v≤nl，1≤u≤nr，保证同一对 v,u 不会出现两次。

输出格式

第一行一个整数，表示幸福程度之和的最大值。

接下来一行 nl 个整数，描述一组最优方案。第 v个整数表示 v 号男生的配偶的编号。如果 v 号男生没配偶请输出 0。

样例一

input

2 2 3
1 1 100
1 2 1
2 1 1

output

100
1 0

限制与约定

1≤nl,nr≤400，1≤m≤160000，1≤w≤1e9。

时间限制：1s

空间限制：256MB

(卡dfs，所以抄了个bfs，留个模板。

参考博客：https://www.cnblogs.com/SilverNebula/p/6710604.html

https://www.cnblogs.com/huyufeifei/p/10350763.html

)

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 4e2+10;
const int INF = 0x3f3f3f3f;
int n,n1,n2,m,N1,N2;
int mp[N][N],slack[N];
int vis1[N],vis2[N];
int ex1[N],ex2[N];
int pre[N],link[N],lx[N],dtime,ans[N];
int q[N*2];
int head,tail;
void update(int x)
{
	if(!x) return ;
	link[x]=pre[x];
	update(lx[pre[x]]);
	lx[pre[x]]=x;
	return ;
}
void bfs(int s)
{
	dtime++;
	int temp;
	for(int i=1;i<=n;i++)
		slack[i]=INF;
	head=tail=0;
	q[tail++]=s;
	while(1)
	{
		while(head<tail)
		{
			int u=q[head++];
			vis1[u]=dtime;
			for(int i=1;i<=n;i++)
			{
				if(vis2[i]^dtime)
				{
					temp=ex1[u]+ex2[i]-mp[u][i];
					if(!temp)
					{
						vis2[i]=dtime;
						pre[i]=u;
						if(!link[i])
						{
							update(i);	
							return ;
						}
						q[tail++]=link[i];		
					}
					else if(slack[i]>temp)
					{
						slack[i]=temp;
						pre[i]=u;
					}
 							
				}	
			}	
		}
		temp=INF;
		for(int i=1;i<=n;i++)
			if(vis2[i]^dtime&&temp>slack[i])
				temp=slack[i];
		for(int i=1;i<=n;i++)
		{
			if(vis1[i]==dtime) ex1[i]-=temp;
			if(vis2[i]==dtime) ex2[i]+=temp;	
			else slack[i]-=temp;	
		}	
		for(int i=1;i<=n;i++)
			if(vis2[i]^dtime&&!slack[i])
			{
				vis2[i]=dtime;
				if(!link[i])
				{
					update(i);
					return ;
				}
				q[tail++]=link[i];
			}
	}
	return ;
}
void KM()
{
	for(int i=1;i<=n;i++)
		bfs(i);
	ll res=0;
	for(int i=1;i<=N2;i++)
		if(mp[link[i]][i])
		{
			ans[link[i]]=i;
			res+=mp[link[i]][i];	
		}
	printf("%lld\n",res);
	for(int i=1;i<=N1;i++)
		printf("%d%c",ans[i],i==N1?'\n':' ');
	return ;
}
int main(void)
{
	
	scanf("%d%d%d",&n1,&n2,&m);
	N1=n1, N2=n2;
	n=max(n1,n2);
	int u,v;
	int w;
	while(m--)
	{
		scanf("%d%d%d",&u,&v,&w);
		mp[u][v]=w;
		if(w>ex1[u])
			ex1[u]=w;
	}
	KM();
		
	
	return 0;	
}