HDU 2448 最短路+KM匹配

先求出每只船到每个港口的最短路，然后建图求KM就行。。。

代码：

//其实在求最大 最小的时候只要用一个模板就行了,把边的权值去相反数即可得到另外一个.求结果的时候再去相反数即可
//邻接矩阵特别需要注意重边的问题，切记
#pragma comment(linker,"/STACK:102400000,102400000")
#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
using namespace std;
typedef long long ll;   //记得必要的时候改成无符号
const int maxn=305;
const int maxm=1000005;
const int INF=1000000000;
int n,nx,ny;
int march[maxn],lx[maxn],ly[maxn],slack[maxn];    //lx,ly为顶标，nx,ny分别为x点集y点集的个数
int w[maxn][maxn];
bool visx[maxn],visy[maxn];

int dfs(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])
            continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0)
        {
            visy[y]=1;
            if(march[y]==-1||dfs(march[y]))
            {
                march[y]=x;
                return 1;
            }
        }
        else if(slack[y]>t)  //不在相等子图中slack 取最小的
            slack[y]=t;
    }
    return 0;
}

int KM()
{
    int i,j;
    memset(march,-1,sizeof(march));
    memset(ly,0,sizeof(ly));
    for(i=1;i<=nx;i++)            //lx初始化为与它关联边中最大的
        for(j=1,lx[i]=-INF;j<=ny;j++)
            if(w[i][j]>lx[i])
               lx[i]=w[i][j];
    for(int x=1;x<=nx;x++)
    {
        for(i=1;i<=ny;i++)slack[i]=INF;
        while(1)
        {
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(dfs(x))     //若成功（找到了增广轨），则该点增广完成，进入下一个点的增广
                break;  //若失败（没有找到增广轨），则需要改变一些点的标号，使得图中可行边的数量增加。
                        //方法为：将所有在增广轨中（就是在增广过程中遍历到）的X方点的标号全部减去一个常数d，
                        //所有在增广轨中的Y方点的标号全部加上一个常数d
            int d=INF;
            for(i=1;i<=ny;i++)
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            for(i=1;i<=nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(i=1;i<=ny;i++)  //修改顶标后，要把所有不在交错树中的Y顶点的slack值都减去d
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
        }
    }
    int res=0;
    for(i=1;i<=ny;i++){
        if(march[i]>-1){
            if(w[march[i]][i]==-INF)continue;
            res+=w[march[i]][i];
        }
    }
    return res;
}

struct EdgeNode
{
    int to;
    int cost;
    int next;
}edge[maxm];
int head[maxn],cnt;
void add(int x,int y,int z)
{
    edge[cnt].to=y;
    edge[cnt].cost=z;
    edge[cnt].next=head[x];
    head[x]=cnt++;
}

void init()
{
    cnt=0;
    memset(head,-1,sizeof(head));
}

int dis[maxn][maxn];
int d[maxn],m;

void DJ(int st)
{
    int x,y,ct;
    typedef pair<int,int>pp;
    priority_queue<pp,vector<pp>,greater<pp> >Q;
    for(int i=1;i<=n+m;i++)d[i]=INF;
    d[st]=0;
    Q.push(pp(0,st));
    while(!Q.empty())
    {
        pp t;
        t=Q.top();
        Q.pop();
        if(t.first>d[t.second])continue;
        if(t.second>m){
            dis[st][t.second]=min(dis[st][t.second],t.first);
            continue;
        }
        x=t.second;
        for(int i=head[x];i!=-1;i=edge[i].next)
        {
            y=edge[i].to; ct=edge[i].cost;
            if(d[x]+ct<d[y]){
                d[y]=d[x]+ct;
                Q.push(pp(d[y],y));
            }
        }
    }
}

int main()
{
    int k,p,i,j;
    int x,y,z;
    int cz[maxn];
    while(~scanf("%d%d%d%d",&n,&m,&k,&p))
    {
        init();
        for(i=1;i<=n;i++)scanf("%d",&cz[i]);
        for(i=1;i<=k;i++){
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        for(i=1;i<=p;i++){
            scanf("%d%d%d",&x,&y,&z);
            add(y,x+m,z);
        }
        for(i=1;i<=n+m;i++)
            for(j=1;j<=n+m;j++)
               dis[i][j]=INF,w[i][j]=-INF;
        for(i=1;i<=n;i++)DJ(cz[i]);
        for(i=1;i<=n;i++){
            x=cz[i];
            for(j=1;j<=n;j++){
                y=j+m;
                if(dis[x][y]!=INF)
                    w[x][y]=-dis[x][y];
            }
        }
        nx=ny=n+m;
        printf("%d\n",-KM());
    }
    return 0;
}