UVA 1349 Optimal Bus Route Design(KM判環)

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=100+5;
const int inf=(1<<30);
int w[maxn][maxn],n;
int lx[maxn],ly[maxn];
int Left[maxn];
bool s[maxn],t[maxn];
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]||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;
    }
}
void KM()
{
    for(int i=1;i<=n;i++){
        Left[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++){
        for(;;)
        {
            for(int j=1;j<=n;j++) s[j]=t[j]=0;
            if(match(i)) break;else update();
        }
    }
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) w[i][j]=-inf;
        for(int i=1;i<=n;i++){
            int j,x;
            while(scanf("%d",&j)&&j)
            {
                scanf("%d",&x);
                w[i][j]=max(w[i][j],-x);//坑點,HDU1853也有這個坑點,兩個點同一方向有多個權值
            }
        }
        KM();
        bool ok=true;
        int ans=0;
        for(int i=1;i<=n;i++){
            ans+=-w[Left[i]][i];
        }
        for(int i=1;i<=n;i++) if(w[Left[i]][i]==-inf) {ok=false;break;}
        if(ok) printf("%d\n",ans);
        else printf("N\n");
    }
    return 0;
}

發佈了690 篇原創文章 · 獲贊 5 · 訪問量 11萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章