HDU 2448 Mining Station on the Sea(Floyd+最优匹配)

HDU 2448 Mining Station on the Sea(Floyd+最优匹配)

http://acm.hdu.edu.cn/showproblem.php?pid=2448

题意:

       给你一个由N个港口和M个海上油田构成的连通图(给出了图中所有的边和权值，即k条无向边连通两个油田，p条有向边连通港口和油田，Notice that once the ship entered the port, it will not come out!),现在给你N个船所在的油田编号,问你让这N条船,每条都回到1个港口去(每个港口最多只能容纳一条船),问你这N条船行走的总距离最短是多少?

分析:

       其实每条船回到任意一个港口去都有一个距离(用Floyd算法算出的最短距离). 建立二分图: 我们把二分图左边放N个港口,右边放N条船,如果第j条船到第i个港口的距离为x,那么就在右j点与左i点之间连一条权值为x的边.

       最终答案即为求 该二分图的最优匹配权值是多少?因为原问题N条船回到N个对应的港口行走总最短距离与现问题二分最优匹配的权值是一一对应的关系.(仔细验证一下,看看是不是一一对应的关系,每一个船只靠岸方案 对应了一个二分图的匹配方案 , 每一个二分图的匹配方案对应了一个船只靠岸方案)

       注意:本题求得是最小花费,而最优匹配求得是最大值.所以我们需要把所有二分图的边权值取负.且港口到油田是单向路径,一定不能添加双向边,因为添加了双向边,会把港口作为中转站.

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300+5;
#define inf 0x3f3f3f3f
//把W[maxn][maxn]数组的内容读进去之后，调用solve(n)即可计算出二分图最优匹配
//不过需要保证该图肯定有完美匹配
//因为本图用的W[][]来表示一个完全图，所以一定存在完美匹配的
struct Max_Match
{
    int W[maxn][maxn],n;  //W是权值矩阵,n为左右点集大小
    int Lx[maxn],Ly[maxn];//左右点集的可行顶标值
    bool S[maxn],T[maxn]; //标记左右点集是否已被访问过
    int left[maxn];       //left[i]=j表右i与左j匹配,为-1时表无匹配

    bool match(int i)
    {
        S[i]=true;
        for(int j=1;j<=n;j++)if(Lx[i]+Ly[j]==W[i][j] && !T[j])
        {
            T[j]=true;
            if(left[j]==-1 || match(left[j]))
            {
                left[j]=i;
                return true;
            }
        }
        return false;
    }

    //更新可行顶标,纳入更多的边进来
    void update()
    {
        int a=1<<30;
        for(int i=1;i<=n;i++)if(S[i])
        for(int j=1;j<=n;j++)if(!T[j])
        {
            a = min(a,Lx[i]+Ly[j]-W[i][j]);
        }
        for(int i=1;i<=n;i++)
        {
            if(S[i]) Lx[i]-=a;
            if(T[i]) Ly[i]+=a;
        }
    }

    int solve(int n)
    {
        this->n=n;
        memset(left,-1,sizeof(left));
        for(int i=1;i<=n;i++)//初始化可行顶标值
        {
            Lx[i]=Ly[i]=0;
            for(int j=1;j<=n;j++)
                Lx[i]=max(Lx[i], W[i][j]);
        }

        for(int i=1;i<=n;i++)
        {
            while(true)
            {
                for(int j=1;j<=n;j++) S[j]=T[j]=false;
                if(match(i)) break;
                else update();
            }
        }

        int ans=0;//最优完美匹配的权值
        for(int i=1;i<=n;i++) ans+= W[left[i]][i];
        return ans;
    }
}KM;
int num[maxn];
int dis[maxn*2][maxn*2];
void floyd(int n)
{
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
	{
		if(dis[i][k]<inf&&dis[k][j]<inf)
		{
			dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
		}
	}
}
int main()
{
	int n,m,k,p;
	int u,v,w;
	while(~scanf("%d%d%d%d",&n,&m,&k,&p))
	{
		//station 1-m
		//ports  m+1~m+n
		memset(dis,inf,sizeof dis);
		for(int i=1;i<=n;i++)
			scanf("%d",&num[i]);
		for(int i=1;i<=k;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			dis[u][v]=dis[v][u]=w;
		}
		for(int i=1;i<=p;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			dis[v][u+m]=w;
		}
		floyd(n+m);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				KM.W[i][j]=-dis[num[i]][j+m];
			}
		}
		printf("%d\n",-KM.solve(n));
	}
}