Hdu 1583（KM算法）

Hdu 1583

 

（1）思路：

考虑一个环中所有的点的入度 = 出度。

所以可以将一个点拆分为两个点，一个入点，一个出点，

一条边相当于建立(u出 -> v入)这样。

所以建立了一个二分图，只要求出二分图的最优完备匹配即可。

所以就用KM算法求解。

 

（2）代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<utility>
using namespace std;
const int maxn = 120;
const int INF = 1e9+10;
int mp[maxn][maxn],vx[maxn],vy[maxn],lx[maxn],ly[maxn],pre[maxn],n,m,cha;
int MIN(int x,int y){
	return x<y?x:y;
}
int MAX(int x,int y){
	return x>y?x:y;
}
bool dfs(int u){
	vx[u] = 1;
	for(int v=1;v<=n;v++)
	if(vy[v]==0&&lx[u]+ly[v]==mp[u][v]){
		vy[v] = 1;
		if(pre[v]==-1||dfs(pre[v])){
			pre[v] = u;
			return true;
		}
	}
	else if(vy[v]==0){
		cha = MIN(cha,lx[u]+ly[v]-mp[u][v]);
	}
	return false;
}
int km(){
	memset(ly,0,sizeof(ly));
	for(int i=1;i<=n;i++){
		lx[i] = mp[i][1];
		for(int j=2;j<=n;j++) lx[i] = MAX(lx[i],mp[i][j]);
	}
	memset(pre,-1,sizeof(pre));
	for(int i=1;i<=n;i++){
		while(true){
			for(int j=1;j<=n;j++) vx[j] = vy[j] = 0;
			cha = INF;
			if(dfs(i)) break;
			for(int j=1;j<=n;j++){
				if(vx[j]==1) lx[j] -= cha;
				if(vy[j]==1) ly[j] += cha;
			}
		}
	}
	int ans = 0;
	for(int i=1;i<=n;i++)
	if(mp[pre[i]][i]!=-INF)
	ans += mp[pre[i]][i];
	else return -1;
	return -ans;
}
int main(void){
	while(~scanf("%d%d",&n,&m)){
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++) mp[i][j] = -INF;
		for(int i=1;i<=m;i++){
			int x,y,z;scanf("%d%d%d",&x,&y,&z);
			mp[x][y] = MAX(-z,mp[x][y]);
		}
		printf("%d\n",km());
	}
	return 0;
}