HDU-3488 Tour

将每个城市拆为入点,出点,然后求一次二分图最小权匹配就是答案

坑点是题目有重边

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=200+10;
const int INF=0x7f7f7f7f;
int nx,ny;
int g[N][N];
int linker[N],lx[N],ly[N];
int slack[N];
bool visx[N],visy[N];
bool dfs(int x)
{
    visx[x]=true;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y]) continue;
        int tmp=lx[x]+ly[y]-g[x][y];
        if(tmp==0)
        {
            visy[y]=true;
            if(linker[y]==-1||dfs(linker[y]))
            {
                linker[y]=x;
                return true;
            }
        }
        else if(slack[y]>tmp)
            slack[y]=tmp;
    }
    return false;
}
int KM()
{
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    for(int i=1;i<=nx;i++)
    {
        lx[i]=-INF;
        for(int j=1;j<=ny;j++)
            if(g[i][j]>lx[i])
                lx[i]=g[i][j];
    }
    for(int x=1;x<=nx;x++)
    {
        for(int j=1;j<=ny;j++)
            slack[j]=INF;
        while(true)
        {
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(dfs(x)) break;
            int d=INF;
            for(int i=1;i<=ny;i++)
                if(!visy[i]&&slack[i]<d)
                    d=slack[i];
            for(int i=1;i<=nx;i++)
                if(visx[i]) lx[i]-=d;
            for(int i=1;i<=ny;i++)
            {
                if(visy[i]) ly[i]+=d;
                else slack[i]-=d;
            }
        }
    }
    int res=0;
    for(int i=1;i<=ny;i++)
        if(linker[i]!=-1)
            res+=g[linker[i]][i];
    return res;
}
int main()
{
    int T;
    int n,m,a,b,c;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        nx=ny=n;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                g[i][j]=-INF;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(a!=b)
            g[a][b]=max(g[a][b],-c);
        }
        printf("%d\n",-KM());
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wl16wzl/article/details/83242197