HDU 5995 Kblack loves flag ---BestCoder Round #90

题目链接

用两个布尔数组分别维护每个行/列是否被插过旗帜，最后枚举每一行、列统计答案即可。空间复杂度O(n+m)，时间复杂度O(n+m+k)。

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int t,ansx[1000005],ansy[1000005],ax,ay;
const int _K=50268147,_B=6082187,_P=100000007;
int _X;
inline int get_rand(int _l,int _r){
    _X=((long long)_K*_X+_B)%_P;
    return _X%(_r-_l+1)+_l;
}
int n,m,k,seed;
int x[1000006],y[1000006];
void Init(){
    scanf("%d%d%d%d",&n,&m,&k,&seed);
    _X=seed;
    for (int i=1;i<=k;++i)
        x[i]=get_rand(1,n),
        y[i]=get_rand(1,m);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        Init();
        memset(ansx,0,sizeof(ansx));
        memset(ansy,0,sizeof(ansy));
        ax=0;ay=0;
        for(int i=1;i<=k;i++)
        {
            ansx[x[i]]=1;
            ansy[y[i]]=1;
        }
        for(int i=1;i<=n;i++)
        if(!ansx[i]) ax++;
        for(int i=1;i<=m;i++)
        if(!ansy[i]) ay++;
        printf("%d %d
",ax,ay);
    }
    return 0;
}