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数组，
$len_{i,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[N][N],p[N],q[N],t[N][N];
char s[N];
bool bz[N];

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;
}