JZOJ3949. 【湖南省队集训2014】Hungry Rabbit

Description

可怕的洪水在夏天不期而至，兔子王国遭遇了前所未有的饥荒，它们不得不去外面的森林里寻找食物。

为了简化起见，我们假设兔子王国中有n 只兔子，编号为1- n。在救济粮到来之前的m 天中，每天恰好有k 只兔子需要去森林里寻找粮食。森林里居住着可怕的大灰狼，所幸兔子已经摸清了大灰狼捕食习惯，即狼们在每一天只会捕食特定编号的兔子。为了安全起见，兔子们需要保证每次出去觅食的k 只兔子都不会被狼捕食。

由于每天出去捕食的兔子都不尽相同，它们为每一天定义了一个生疏度pi ，即第i 天出来寻找食物，但是第i -1 天却没有出来觅食的兔子个数。规定第1 天的生疏度为0.

现在兔子们希望在保证安全的前提下，每天的生疏度不能超过L，请为兔子们构造一个合法的方案。

Input

第一行包括四个整数n, m, k 和L.

接下来n 行，每行一个长度为m 的01 串。其中第i 行第j 个字符若为0，则表示狼在第j 天会捕食编号为i 的兔子，为1 则表示不捕食。

Output

m 行，每行k 个1 - n 之间互不相同的整数，代表这一天出去寻找食物的兔子编号。如果没有合法方案，则输出一行-1 即可。

Sample Input

5 4 3 1

1001

1101

1111

1110

0111

Sample Output

2 3 4

2 3 4

3 4 5

2 3 5

【样例解释】

对于样例，在这4 天中，出去觅食的兔子集合分别为{2,3,4}，{2,3,4}，{3,4,5}，{2,3,5}。

Data Constraint

• 对于20% 的测试数据，1 <= n,m <=10

• 对于100% 的测试数据，1<= n,m <=800， 1<= k<=n， 1<=L<= k

题解

第一眼就感觉是网络流，

虽然网络流可以做，但是有更简单的方法。

一种贪心的思想，

如果某只兔子可以连续出去的天数更多，

就应该选择这只兔子尽量多的天，

这样需要更换兔子的兔子次数就会变少。

先预处理出len数组，

leni,jleni,j表示从第i天开始，j这只兔子最多可能连续多少天外出。

在第i-1天兔子外出的集合为q，那先求出第i天可以出去的，且不在q集合里面的兔子，

根据他们可以连续出去的天数来排序，

与原来q中的兔子比较，

如果更优就用p中的兔子来替换q中的兔子。

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <time.h>
#define ll long long
#define N 803
#define M 103
#define db double
#define P putchar
#define G getchar
#define inf 998244353
using namespace std;
char ch;
void read(int &n)
{
n=0;
ch=G();
while((ch<'0' || ch>'9') && ch!='-')ch=G();
ll w=1;
if(ch=='-')w=-1,ch=G();
while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

int tot,ans;
int n,m,k,l,id,len

,p
,q
,t

;
char s
;
bool bz
;

bool cmp(int x,int y)
{
return len[id][x]>len[id][y];
}

int main()
{
read(n);read(m);read(k);read(l);
for(int i=1;i<=n;i++)
{
scanf("%s",s+1);
for(int j=m;j;j--)
len[j][i]=s[j]=='0'?0:(len[j+1][i]+1);
}

for(id=1;id<=m;id++)
{
tot=0;
memset(bz,1,sizeof(bz));
for(int i=1;i<=k;i++)
bz[q[i]]=0;
for(int i=1;i<=n;i++)
if(bz[i] && len[id][i])p[++tot]=i;
sort(p+1,p+1+tot,cmp);
sort(q+1,q+1+k,cmp);

if(id==1)
{
if(tot<k)
{
P('-');P('1');
return 0;
}
for(int i=1;i<=k;i++)
t[id][i]=p[i];

swap(q,p);
continue;
}

for(int i=1;i<=l && i<=tot;i++)
if(len[id][p[i]]>len[id][q[k-i+1]])swap(p[i],q[k-i+1]);
if(len[id][q[k]]==0)
{
P('-');P('1');
return 0;
}

for(int i=1;i<=k;i++)
t[id][i]=q[i];
}

for(int i=1;i<=m;i++)
{
for(int j=1;j<=k;j++)
write(t[i][j]),P(' ');
P('\n');
}

return 0;
}