题目大意:给定一个棋盘上放有各种颜色的气球,对于每种颜色的气球,每次操作可以选择一个该种颜色的气球以及与改气球同一行同一列同一种颜色的气球,将他们全部扎破,如果操作次数大于k就输出该颜色。
题解:我们对每种颜色的气球进行讨论,将每种颜色气球根据行与列建立二分图,如果最大匹配数大于k就输出(注意输出的格式有些坑,详见代码)
#include<bits/stdc++.h>
using namespace std;
int n, m, k;
const int maxn = 105;
int edge[maxn][maxn], vis[maxn], link[maxn]; //vis[]数组我称为临时预定数组,vis[j] = a表示一轮模拟匹配中,女孩j被男孩a预定了。
int color[52];
int nowcolor;
vector<int>ans;
int dfs(int l)
{
for (int r = 1; r <= n; r++)
{
if (edge[l][r]==nowcolor && !vis[r])
{
vis[r] = 1;
if (link[r] == -1 || dfs(link[r]))
{
link[r] = l;
return 1;
}
}
}
return 0;
}
int hungary(int color)
{
int ans = 0;
nowcolor = color;
memset(link, -1, sizeof(link));
for (int i = 1; i <= n; i++)
{
memset(vis, 0, sizeof(vis));
if (dfs(i))ans++;
}
return ans;
}
int main()
{
int T = 1;
while (~scanf_s("%d%d",&n,&k)&&n&&k)
{
memset(edge, 0, sizeof(edge));
memset(color, 0, sizeof(color));
ans.clear();
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
{
cin >> edge[i][j];
color[edge[i][j]] = 1;
}
}
for (int i = 1; i <= 50; i++)
{
if (color[i])
{
if (hungary(i) > k)ans.push_back(i);
}
}
if (ans.size() == 0)cout <<-1<< endl;
else
{
for (int i = 0; i < ans.size(); ++i) {
if (i) printf(" ");
printf("%d", ans[i]);
}
cout << endl;
}
}
}