50 years, 50 colors HDU

和棋盘游戏一个类型的题目

矩阵转换为二分图最大匹配问题   每次碰撞气球都是一行或一列

#include<bits/stdc++.h>
using namespace std;
#define MAXI 105
int mp[MAXI][MAXI];
int used[MAXI];
int vis[MAXI];
int n,m;
int color[];
bool dfs(int  x,int color)
{
    for(int j=;j<=n;j++)
    {
        if(mp[x][j]==color&&!used[j])
        {
            used[j]=;
            if(!vis[j]||dfs(vis[j],color))
              {
                  vis[j]=x;
                  return true;
              }
        }
    }
    return false;
}

int find1(int x)
{    int ans=;
 memset(vis,,sizeof(vis));
     for(int i=;i<=n;i++)
       {
           memset(used,,sizeof(used));
           if(dfs(i,x))ans++;
       }
       return ans;
}

int main()
{
  int k;
   while(scanf("%d%d",&n,&k)==&&(n||k))
   {
       int ans[MAXI];
       memset(color,,sizeof(color));
       memset(ans,,sizeof(ans));

        for(int i=;i<=n;i++)
        for(int j=;j<=n;j++)
         {
           scanf("%d",&mp[i][j]);
           color[mp[i][j]]=;
         }

       int n=;
       for(int i=;i<=;i++)
       {
           if(color[i])
           {
               if(k<find1(i))
                   ans[n++]=i;
           }
       }

       sort(ans,ans+n);

       if(n==)printf("-1\n");
       else
        {
         for(int i=;i<n-;i++)
           printf("%d ",ans[i]);
           printf("%d\n",ans[n-]);
        }

   }
  return ;
}