题目概述
给出一个n*m的矩阵(n<=m),选出n个数,使这n个数均不在同一行且不在同一列,设选择的数中第K大为Kth,求 Kthmin 。
题目概述
直接暴力枚举所有选择情况肯定不行的。我们会想到枚举第K大的数now,然后再通过某种方法验证能否有一种选择方法使选出的小于等于now的个数为n-K+1,顺着这种思路想下去,就会发现可以使用二分枚举(答案一定是矩阵内某个数,否则肯定不是最优解),但是验证成功的条件应该改为选出的小于等于now的个数大于等于n-K+1。
但是最主要的问题就是如何求出最优选择方法。观察题目要求,每行每列都只能放一个,这种类型的条件是经典的二分图匹配。对于每一个小于等于now的数(i,j),都建一条i->j的边,然后刷二分图最大匹配就行了。
示例程序
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=250,maxm=250,MAXINT=((1<<30)-1)*2+1;
int n,m,K,pic[maxn+5][maxm+5];
int E,son[maxn*maxm+5],nxt[maxn*maxm+5],lnk[maxn+5],who[maxn+5];
bool vis[maxn+5];
void Add(int x,int y) {son[++E]=y;nxt[E]=lnk[x];lnk[x]=E;}
bool Find(int x)
{
if (vis[x]) return false;vis[x]=true;
for (int j=lnk[x];j;j=nxt[j])
if (!who[son[j]]||Find(who[son[j]])) {who[son[j]]=x;return true;}
return false;
}
bool check(int MIN)
{
memset(lnk,0,sizeof(lnk));E=0;
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
if (pic[i][j]<=MIN) Add(i,j);
int MAX=0;
memset(who,0,sizeof(who));
for (int i=1;i<=n;i++)
{
memset(vis,0,sizeof(vis));
MAX+=Find(i);
}
return MAX>=K;
}
int main()
{
freopen("program.in","r",stdin);
freopen("program.out","w",stdout);
scanf("%d%d%d",&n,&m,&K);K=n-K+1;
int L=MAXINT,R=-MAXINT;
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
scanf("%d",&pic[i][j]),L=min(L,pic[i][j]),R=max(R,pic[i][j]);
while (L<=R)
{
int mid=L+(R-L>>1);
if (check(mid)) R=mid-1; else L=mid+1;
}
printf("%d\n",L);
return 0;
}