HDU 1498 50 years, 50 colors(匈牙利算法 二分最大匹配數)

題意:給出一個矩陣 ,裏面放一些氣球(用數字表示),然後給了一個次數,一個人可以射k次,每次選一行或者一列,一行貨一列中相同顏色的氣球(相同數字)可以一下全射爆,問一個人射了k次後矩陣中哪些氣球不能被射爆,全被射爆輸出-1,如果剩餘多個氣球,按升序輸出(注意看題..這個我一開始都沒看到)


這也是個匈牙利算法求最大匹配數的,不過中間要加一些東西,具體看代碼吧

題目鏈接:點擊打開鏈接

<strong><span style="font-size:18px;color:#ff6600;">
</span></strong>
<strong><span style="font-size:18px;color:#ff6600;">代碼註釋:</span></strong>
<strong><span style="font-size:18px;color:#ff6600;">
</span></strong>
<pre name="code" class="cpp"><strong><span style="font-size:18px;color:#ff6600;">#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;

#define N 105
int a[N][N],visit[N],link[N],use[N];
int n;

int dfs(int i,int color)//這裏稍有不同,加了一個傳遞值,傳遞的權值
{
    for(int j=1; j<=n; j++)
    {
        if(a[i][j]==color&&!visit[j])
        {
            visit[j]=1;
            if(link[j]==-1||dfs(link[j],color))
            {
                link[j]=i;
                return 1;
            }
        }
    }
    return 0;
}

int main()
{
    int i,j,k,m;
    int ans[N];
    while(cin>>n>>k,n+k)
    {
        m=0;
        memset(a,0,sizeof(a));
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
                cin>>a[i][j];
        memset(use,0,sizeof(use));
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
            {
                memset(link,-1,sizeof(link));
                int  sum=0;
                if(!use[a[i][j]])
                {
                    use[a[i][j]]=1;
                    for(int t=1; t<=n; t++)
                    {
                        memset(visit,0,sizeof(visit));
                        sum+=dfs(t,a[i][j]);
                    }
                }
                if(sum>k) //如果射爆所有氣球需要的次數比給的限制次數大,就輸出該顏色氣球
                    ans[m++]=a[i][j];
            }
        if(m==0)
            cout<<-1<<endl;
        else
        {
            sort(ans,ans+m);//記得升序輸出結果
            for(i=0; i<m-1; i++)
                cout<<ans[i]<<" ";
            cout<<ans[i]<<endl;
        }
    }
}</span></strong>




發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章