hdu2426 Interesting Housing Problem

最新推荐文章于 2019-12-18 21:58:47 发布

别动我的白羊毛

最新推荐文章于 2019-12-18 21:58:47 发布

阅读量202

点赞数

分类专栏：二分图匹配

本文链接：https://blog.csdn.net/yz467796454/article/details/78116350

版权

二分图匹配专栏收录该内容

9 篇文章 0 订阅

订阅专栏

之前用的那个模板是完全匹配，然后点的个数都用的N，这题有N,M，自己改了一下板子，结果改错了，索性换一个

这个板子来自http://www.cnblogs.com/someblue/p/4027628.html 是hdu1853的题解的代码修改得到的

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#include <map>
#include <utility>
#include <queue>
#include <stack>
using namespace std;

const int INF=1e9;
const double eps=1e-6;
const int N = 550;

int nx,ny;
int g[N][N];
int linker[N],lx[N],ly[N];
int slack[N];
int visx[N],visy[N];

int n,m;

bool DFS(int x)
{
    visx[x]=true;
    for(int y=0;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=0;i<nx;i++)
    {
        lx[i]=-INF;
        for(int j=0;j<ny;j++)
            if(g[i][j]>lx[i])
                lx[i]=g[i][j];
    }
    for(int x=0;x<nx;x++)
    {
        for(int i=0;i<ny;i++)
            slack[i]=INF;
        while(true)
        {
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(DFS(x)) break;
            int d = INF;
            for(int i=0;i<ny;i++)
                if(!visy[i] && d>slack[i])
                    d=slack[i];
            for(int i=0;i<nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(int i=0;i<ny;i++)
            {
                if(visy[i]) ly[i]+=d;
                else slack[i]-=d;
            }
        }
    }
    int res = 0, cnt = 0;
    for(int i=0;i<ny;i++)
    {
        if(linker[i]==-1 || g[linker[i]][i]==-INF)
            continue;
        res += g[linker[i]][i];
        cnt++;
    }
    if(cnt!=nx) return -1;
    return res;
}

int main(){
	int i,j,e,c=1;
	while(scanf("%d%d%d",&nx,&ny,&e)!=EOF){
		for(i=0;i<nx;i++){
			for(j=0;j<ny;j++){
				g[i][j]=-INF;
			}
		}
		for(i=0;i<e;i++){
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			if(c<0)continue;
			if(c>g[a][b]){
				g[a][b]=c;
			}
		}
		printf("Case %d: %d\n",c++,KM());
	}
	return 0;
}