第二次周赛D题 hdu 2448 【最短路径+最佳匹配】

【第二次周赛D题】

 

【原题链接】

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

【题目大意】

输入一幅图，图中海上若干油井，岸边有若干港口，可以航行的油井之间有相互距离，港口和油井之间有相互距离（港口数小于油井数），有几艘轮船，每个在一个油井（轮船数等于港口数），问所有轮船到港口的所需走的距离和最小是多少？

【解题思想】

先用dij算法找到所有港口到有船油井的最短距离，再通过最佳匹配，找到和值最小的有船油井和港口的搭配。

【注意】

输入量大，需要用scanf输入！

 

[code]

#include <iostream>

#include <cstdio>

#define MAXN 310

#define inf 1000000000

#define _clr(x) memset(x,0xff,sizeof(long)*n)

using namespace std;

 

void dijkstra(long n,long mat[][MAXN],long s,long* MIN){

    long v[MAXN],i,j,k;

    for (i=0;i<n;i++)

        MIN[i]=inf,v[i]=0;

    for (MIN[s]=0,j=0;j<n;j++){

        for (k=-1,i=0;i<n;i++)

            if (!v[i]&&(k==-1||MIN[i]<MIN[k]))

                k=i;

        for (v[k]=1,i=0;i<n;i++)

            if (!v[i]&&MIN[k]+mat[k][i]<MIN[i])

                MIN[i]=MIN[k]+mat[k][i];

    }

}

 

long kuhn_munkras(long m,long n,long mat[][MAXN],long* match1,long* match2){

    long s[MAXN],t[MAXN],l1[MAXN],l2[MAXN],p,q,ret=0,i,j,k;

    for (i=0;i<m;i++)

        for (l1[i]=-inf,j=0;j<n;j++)

            l1[i]=mat[i][j]>l1[i]?mat[i][j]:l1[i];

    for (i=0;i<n;l2[i++]=0);

    for (_clr(match1),_clr(match2),i=0;i<m;i++){

        for (_clr(t),s[p=q=0]=i;p<=q&&match1[i]<0;p++)

            for (k=s[p],j=0;j<n&&match1[i]<0;j++)

                if (l1[k]+l2[j]==mat[k][j]&&t[j]<0){

                    s[++q]=match2[j],t[j]=k;

                    if (s[q]<0)

                        for (p=j;p>=0;j=p)

                            match2[j]=k=t[j],p=match1[k],match1[k]=j;

                }

        if (match1[i]<0){

            for (i--,p=inf,k=0;k<=q;k++)

                for (j=0;j<n;j++)

                    if (t[j]<0&&l1[s[k]]+l2[j]-mat[s[k]][j]<p)

                        p=l1[s[k]]+l2[j]-mat[s[k]][j];

            for (j=0;j<n;l2[j]+=t[j]<0?0:p,j++);

            for (k=0;k<=q;l1[s[k++]]-=p);

        }

    }

    for (i=0;i<m;i++)

        ret+=mat[i][match1[i]];

    return ret;

}

 

long sMap[310][310],map[310][310],partMap[310][310],MIN[310],match1[310],match2[310],in[310];;

 

int main()

{

    long n,m,k,p,i,j,inStation,from,to,d;

    while(scanf("%ld%ld%ld%ld",&n,&m,&k,&p)!=EOF)

    {

        for(i=0;i<n;i++)

        {

            scanf("%ld",&inStation);

            in[i]=inStation;

        }

        for(i=0;i<=m;i++)

            for(j=0;j<=m;j++) sMap[i][j]=inf;

        for(i=0;i<k;i++)

        {

            scanf("%ld%ld%ld",&from,&to,&d);

            sMap[from][to]=d;

            sMap[to][from]=d;

        }

        for(i=0;i<n+m;i++)

            for(j=0;j<n+m;j++)

            {

                map[i][j]=inf;

                partMap[i][j]=inf;

            }

        for(i=0;i<p;i++)

        {

            scanf("%ld%ld%ld",&from,&to,&d);

            to+=n;

            map[from][to]=d;

            map[to][from]=d;

        }

        for(i=0;i<n;i++)

        {

            for(j=0;j<=m;j++)

            {

                sMap[0][j]=map[i+1][j+n];

                sMap[j][0]=map[i+1][j+n];

            }

            dijkstra(m+1,sMap,0,MIN);

            for(j=0;j<n;j++)

            {

                partMap[i][j]=-MIN[in[j]];

            }

        }

        cout<<-kuhn_munkras(n,n,partMap,match1,match2)<<endl;

    }

    return 0;

}

[\code]