题目链接:HDU-1498
主要思路:
对于每一个当前颜色的气球,我们有两种方式将它引爆,一是它所在的行被引爆(可能是引爆它,也可能是引爆与它同行的相同颜色的气球),二是它所在的列被引爆(注意事项同上)。这两种方式在引爆一个气球时只会被执行一种。故建立二分图,求行与列的最大匹配即可。
AC代码:
#include<cstdio>
#include<cstring>
#include<vector>
#define M 105
using namespace std;
vector<int>way[55][M];
int mp[M][M];
bool used[105];
int chose[105];
bool mark[M];
bool find(int col,int now) {//匈牙利求最大匹配
for(int i=0; i<way[col][now].size(); i++) {
int nxt=way[col][now][i];
if(!used[nxt]) {
used[nxt]=1;
if(!chose[nxt]||find(col,chose[nxt])) {
chose[nxt]=now;
return 1;
}
}
}
return 0;
}
vector<int>ANS;
int main() {
int n,k;
while(~scanf("%d%d",&n,&k)) {
memset(mark,0,sizeof(mark));
if(!n&&!k)return 0;
for(int i=1; i<=n; i++) {
for(int j=1; j<=n; j++) {
scanf("%d",&mp[i][j]);
mark[mp[i][j]]=1;//标记该颜色有没有出现
}
}
for(int i=1; i<=n; i++) {
for(int j=1; j<=n; j++) {
way[mp[i][j]][i].push_back(j);//行连列
}
}
ANS.clear();
for(int j=1; j<=50; j++) {
if(!mark[j])continue;//对于每种颜色求最大匹配
memset(chose,0,sizeof(chose));
int ans=0;
for(int i=1; i<=n; i++) {
memset(used,0,sizeof(used));
if(find(j,i))ans++;
}
if(ans>k)ANS.push_back(j);//若大于k就加入ans
}
for(int i=1;i<=50;i++){
if(!mark[i])continue;
for(int j=1;j<=n;j++)way[i][j].clear();
}
if(ANS.size()!=0) {
for(int i=0; i<ANS.size()-1; i++)printf("%d ",ANS[i]);
printf("%d\n",ANS[ANS.size()-1]);
} else puts("-1");
}
}