洛谷P3622 [APIO2007]动物园 题解

题目链接

分析：

因为小朋友是否高兴取决于动物的在或不在，很容易想到状压DP。看一眼数据范围：$n⩽10^4,c⩽5\times 10^4$，似乎又不太可能。但经过仔细观察，我们注意到一句话：

每个小朋友站在大围栏圈的外面，可以看到连续的 $5$ 个围栏

发现每个小朋友是否高兴仅与他所能看到的$5$个围栏有关，而$5$非常小，可以拿来压缩！

那么我们可以设计出状态：令$f_{i,S}$表示从第$i$位开始的五个动物是否留下的状态为$S$时前$i+4$个动物最多能使多少小朋友高兴，其中
$0⩽S⩽11111_{(2)}=31$，二进制下低位对应编号小的动物，对应位为$1$则代表这个动物留下来，否则代表撤走。

考虑转移方程前，先思考一件事：当第$i$位到第$i+4$位的状态为$S$时，从第$i-1$位开始的五位的状态$S^{\prime }$可能是什么？

因为第$i$位到第$i+3$这四位的状态是共有的，也是确定的，所以我们可以先把这四位处理出来:$S\&(01111_{(2)}=15)$，这样可以直接把第$i+4$位处理掉，又因为第$i$位到第$i+3$这四位在$S^{\prime }$中是后四位，所以我们还需要左移一位：$(S\&15)<<1$，这里要注意位运算的优先级，然后就是讨论第$i-1$位了，当第$i-1$位为$0$时，有$S^{\prime }=(S\&15)<<1$；当第$i-1$位为$1$时，则为$S^{\prime }=(S\&15)<<1|1$。

于是就可以写转移方程了：
$$f_{i,S}=max(f_{i-1,(S\&15)<<1},f_{i-1,(S\&15)<<1|1})+g_{i,S}$$

其中$g_{i,S}$表示从第$i$位开始的五个动物是否留下的状态为$S$时能使多少在第$i$位的小朋友高兴，显然可以预处理出来

写的时候要注意初始化问题，我们要选择一个起点并枚举从起点开始的五位的状态，并把当前枚举到的状态设为0（最后再次循环到起点时再计算这五位的贡献，方便计算），其它状态全部设为负无穷。考虑到起点会循环到两遍，为了防止下标取模出现问题、简化代码，可以选择第$1$位的前一位（即$n$）为起点，这样初始化时第一维就可以选择$0$（$0$在$1$的前一位，而转移时都是由前一位更新后一位），循环时也就只会从$1$到$n$，不存在取模问题了。

详见代码

Code:

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 5e4 + 50,maxs = 31,inf = 1e9;
int n,m,a,b,c,t,p,q,ans,f[maxn][maxs],g[maxn][maxs];//m即为题目中的c
int read(){
    int x = 0;
    char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    while(c >= '0' && c <= '9') x = x * 10 + (c ^ 48),c = getchar();
    return x;
}
int main(){
    n = read(),m = read();
    for(int i = 1; i <= m; i ++){//预处理出g
        a = read(),b = read(),c = read(),p = 0,q = 0;//p为当前小朋友害怕的动物的状态，q为喜欢的
        for(int j = 1; j <= b; j ++) t = (read() - a + n) % n,p |= (1 << t);//t为当前动物与小朋友的相对位置，下同
        for(int j = 1; j <= c; j ++) t = (read() - a + n) % n,q |= (1 << t);
        for(int j = 0; j <= maxs; j ++) if((~j & p) || (j & q)) g[a][j] ++;
        //如果p中有j没有的动物或者q中有j中有的动物，则状态为j时当前小朋友开心，g[a][j]应++
    }
    for(int P = 0; P <= maxs; P ++){//枚举第n位开始的五位的状态
        for(int S = 0; S <= maxs; S ++) f[0][S] = -inf;
        f[0][P] = 0;//只有f[0][P]能达到
        for(int i = 1; i <= n; i ++)
            for(int S = 0; S < maxs; S ++)
                f[i][S] = max(f[i - 1][(S & 15) << 1],f[i - 1][(S & 15) << 1 | 1]) + g[i][S];
        ans = max(ans,f[n][P]);//统计最大值
    }
    cout << ans << endl;
    return 0;
}