APIO2007 洛谷P3622 动物园

题目传送门：https://www.luogu.org/problemnew/show/P3622

题解：首先，对于每个小朋友，对他有影响的只有能看到的5个点，故想到状压。

变量声明：num[i][j]:第[i,i+4]有状态为j的满足人数

状态转移方程：f[i][j]=max(f[i-1][(j&15)<<1],f[i-1][(j&15)<<1|1])+num[i][j]//取出前4个，枚举第i-1个是否有取

要注意，因为原题是一个环，故到n的状态要与初始状态相同。

代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
#define MN 50005
using namespace std;
int n,m,num[MN][40],f[MN][40],ans;
void init()
{
    
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int a,b,c,l=0,d=0,t;
        scanf("%d%d%d",&a,&b,&c);
        for(int j=1;j<=b;j++){
            scanf("%d",&t);
            t=(t-a+n)%n;
            l|=1<<t;
        }
        for(int j=1;j<=c;j++){
            scanf("%d",&t);
            t=(t-a+n)%n;
            d|=1<<t;
        }
        for(int j=0;j<32;j++)
            if((j&l)||(~j&d)) num[a][j]++;
    }
}
void solve(){
    for(int i=0;i<32;i++){
        memset(f[0],128,sizeof(f[0]));
        f[0][i]=0;
        for(int j=1;j<=n;j++)
            for(int k=0;k<32;k++)
                f[j][k]=max(f[j-1][(k&15)<<1],f[j-1][(k&15)<<1|1])+num[j][k];
        ans=max(ans,f[n][i]);
    }
    printf("%d",ans);
}
int main()
{
    init();
    solve();
    return 0;
}