hdu1533(KM算法逆向版)

思路:

原版KM算法求的是最大权值完美匹配,这题要求的是最小权值完美匹配,第一种思路,把边取负,跑最大权值km算法,得到的答案取负即可。

代码:

#include <bits/stdc++.h>

using namespace std;
const int maxn = 103;
const int inf = 0x3f3f3f3f;
int w[maxn][maxn],lx[maxn],ly[maxn],slack[maxn],link[maxn],visx[maxn],visy[maxn],arr[maxn][maxn];
int n,m;

bool dfs(int u)
{
    visx[u] = 1;
    for(int v = 1;v<=n;v++)
    {
        if(!visy[v])
        {
            int gap = lx[u]+ly[v]-w[u][v];
            if(gap==0)
            {
                visy[v] = 1;
                if(link[v]==-1||dfs(link[v]))
                {
                    link[v] = u;
                    return true;
                }
            }
            else
            {
                slack[v] = min(slack[v],gap);
            }
        }
    }
    return false;
}

void update()
{
    int d = inf;
    for(int i = 1;i<=n;i++)
    {
        if(!visy[i])
        {
            d = min(d,slack[i]);
        }
    }
    for(int i = 1;i<=n;i++)
    {
        if(visx[i])
            lx[i] -= d;
        if(visy[i])
            ly[i] += d;
    }
}

int km()
{
    memset(link,-1,sizeof(link));
    for(int i = 1;i<=n;i++)
    {
        ly[i] = 0,lx[i] = -inf;
        for(int j = 1;j<=n;j++)
        {
            //cout<<w[i][j]<<endl;
            lx[i] = max(lx[i],w[i][j]);
        }
        //cout<<lx[i]<<endl;
    }
    for(int i = 1;i<=n;i++)
    {
        memset(slack,inf,sizeof(slack));
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(i))
                break;
            else
                update();
        }
    }
    int ans = 0;
    for(int i = 1;i<=n;i++)
    {
        ans += lx[i]+ly[i];
    }
    return ans;
}

struct node
{
    int x,y;
}px[maxn],py[maxn];

int xl,yl;

void input()
{
    xl = yl = 1;
    for(int i = 0;i<n;i++)
    {
        getchar();
        for(int j = 0;j<m;j++)
        {
            scanf("%c",&arr[i][j]);
        }
    }
    for(int i = 0;i<n;i++)
    {
        for(int j = 0;j<m;j++)
        {
            if(arr[i][j]=='m')
            {
                px[xl].x = i;
                px[xl].y = j;
                xl++;
            }
            if(arr[i][j]=='H')
            {
                py[yl].x = i;
                py[yl].y = j;
                yl++;
            }
        }
    }
    //cout<<xl<<endl<<yl<<endl;
}

int main()
{
    while(~scanf("%d%d",&n,&m)&&n&&m)
    {
        input();
        for(int i = 1;i<xl;i++)
        {
            for(int j = 1;j<yl;j++)
            {
                int dis = abs(px[i].x-py[j].x)+abs(px[i].y-py[j].y);
                w[i][j] = -dis;
                //cout<<-dis<<endl;
            }
        }
        n = xl-1;
        printf("%d\n",-km());
    }
    return 0;
}


第二种思路:

在深入理解了KM算法的修改顶标数组的操作后,可以通过变更修改操作,使得求出的是最小权值。。

原版的KM算法,两个点的顶标值和是大于等于这两点的边的权值的,所以相等子图存在的一定是权值大的那部分边,,

那么如果两点顶标的值的和小于等于这两点的边的权值,那么相等子图存在的一定是权值小的那部分边,

所以slack数组的值仍然是顶标和与边权值的绝对值,,每次更新同样取最小的slack值d。

更新顶标时,注意lx[i]这时要+=d(新加入的边权值是增大的),ly[i]-=d;

代码:

#include <bits/stdc++.h>

using namespace std;
const int maxn = 103;
const int inf = 0x3f3f3f3f;
int w[maxn][maxn],lx[maxn],ly[maxn],slack[maxn],link[maxn],visx[maxn],visy[maxn],arr[maxn][maxn];
int n,m;

bool dfs(int u)
{
    visx[u] = 1;
    for(int v = 1;v<=n;v++)
    {
        if(!visy[v])
        {
            int gap = -lx[u]-ly[v]+w[u][v];//原版的是lx[u]+ly[v]-w[u][v]
            if(gap==0)
            {
                visy[v] = 1;
                if(link[v]==-1||dfs(link[v]))
                {
                    link[v] = u;
                    return true;
                }
            }
            else
            {
                slack[v] = min(slack[v],gap);
            }
        }
    }
    return false;
}

void update()
{
    int d = inf;
    for(int i = 1;i<=n;i++)
    {
        if(!visy[i])
        {
            d = min(d,slack[i]);
        }
    }
    for(int i = 1;i<=n;i++)
    {
        if(visx[i])
            lx[i] += d;
        if(visy[i])
            ly[i] -= d;
    }//原版为lx[i]-=d,ly[i]+=d;
}

int km()
{
    memset(link,-1,sizeof(link));
    for(int i = 1;i<=n;i++)
    {
        ly[i] = 0,lx[i] = inf;
        for(int j = 1;j<=n;j++)
        {
            //cout<<w[i][j]<<endl;
            lx[i] = min(lx[i],w[i][j]);
        }
        //cout<<lx[i]<<endl;
    }
    for(int i = 1;i<=n;i++)
    {
        memset(slack,inf,sizeof(slack));
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(i))
                break;
            else
                update();
        }
    }
    int ans = 0;
    for(int i = 1;i<=n;i++)
    {
        ans += lx[i]+ly[i];
    }
    return ans;
}

struct node
{
    int x,y;
}px[maxn],py[maxn];

int xl,yl;

void input()
{
    xl = yl = 1;
    for(int i = 0;i<n;i++)
    {
        getchar();
        for(int j = 0;j<m;j++)
        {
            scanf("%c",&arr[i][j]);
        }
    }
    for(int i = 0;i<n;i++)
    {
        for(int j = 0;j<m;j++)
        {
            if(arr[i][j]=='m')
            {
                px[xl].x = i;
                px[xl].y = j;
                xl++;
            }
            if(arr[i][j]=='H')
            {
                py[yl].x = i;
                py[yl].y = j;
                yl++;
            }
        }
    }
    //cout<<xl<<endl<<yl<<endl;
}

int main()
{
    while(~scanf("%d%d",&n,&m)&&n&&m)
    {
        input();
        for(int i = 1;i<xl;i++)
        {
            for(int j = 1;j<yl;j++)
            {
                int dis = abs(px[i].x-py[j].x)+abs(px[i].y-py[j].y);
                w[i][j] = dis;
                //cout<<dis<<endl;
            }
        }
        n = xl-1;
        printf("%d\n",km());
    }
    return 0;
}


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值