- 看题意就会知道这个是二分匹配模型,转化一下,将行数变成X集合,列数变成Y集合,某一点(i,j)为1的话,将i和j连一条边
- 所以求的是最小点覆盖集,等于最大匹配数
- 最后是找出点覆盖集,对于每一条边,肯定有一点属于覆盖记得,所以对于边i-j,如果j没有匹配,那么i肯定是答案,之后再循环一遍,将所有的结果保存并输出
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int MAXN=3000+5;
const int INF=1<<28;
int g[MAXN][MAXN],Mx[MAXN],My[MAXN],Nx,Ny;
int dx[MAXN],dy[MAXN],dis;
bool vst[MAXN];
bool searchP()
{
queue<int>Q;
dis=INF;
memset(dx,-1,sizeof(dx));
memset(dy,-1,sizeof(dy));
for(int i=1;i<=Nx;i++)
if(Mx[i]==-1)
{
Q.push(i);
dx[i]=0;
}
while(!Q.empty())
{
int u=Q.front();
Q.pop();
if(dx[u]>dis) break;
for(int v=1;v<=Ny;v++)
if(g[u][v]&&dy[v]==-1)
{
dy[v]=dx[u]+1;
if(My[v]==-1) dis=dy[v];
else
{
dx[My[v]]=dy[v]+1;
Q.push(My[v]);
}
}
}
return dis!=INF;
}
bool DFS(int u)
{
for(int v=1;v<=Ny;v++)
if(!vst[v]&&g[u][v]&&dy[v]==dx[u]+1)
{
vst[v]=1;
if(My[v]!=-1&&dy[v]==dis) continue;
if(My[v]==-1||DFS(My[v]))
{
My[v]=u;
Mx[u]=v;
return 1;
}
}
return 0;
}
int MaxMatch()
{
int res=0;
memset(Mx,-1,sizeof(Mx));
memset(My,-1,sizeof(My));
while(searchP())
{
memset(vst,0,sizeof(vst));
for(int i=1;i<=Nx;i++)
if(Mx[i]==-1&&DFS(i)) res++;
}
return res;
}
char s[1024];
int main()
{
int p,n,m,i,j,t;
while(~scanf("%d%d%d",&n,&m,&p))
{
memset(g,0,sizeof(g));
for(i=0;i<n;i++)
{
scanf("%s",s);
for(j=0;j<m;j++)
g[i+1][j+1]=s[j]-'0';
}
Nx=n;
Ny=m;
t=MaxMatch();
if(t>p) printf("NOT ENOUGH TANK\n");
else
{
printf("%d\n",t);
int r[1024],c[1024];
memset(r,0,sizeof(r));
memset(c,0,sizeof(c));
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
{
if(g[i][j]&&My[j]==-1)
r[i]=1;
}
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
{
if(g[i][j])
{
if(!r[i])
c[j]=1;
}
}
printf("ROW:");
for(i=1;i<=n;i++)
if(r[i]) printf(" %d",i);
printf("\n");
printf("COLUMN:");
for(i=1;i<=m;i++)
if(c[i]) printf(" %d",i);
printf("\n");
}
}
return 0;
}