背包+bitset

题目描述

题解：（参考于此）

这个题是个典型的分组背包。
f[i][j] 为前i个数能否构成 j ，考虑第i个数选还是不选。
显然 f [ i-1 ][ j-x[i]²]==1的话f[i][j]就是可以的，x[i]的取值范围是 l[i] 到 r[i]。

这样的话其实有点浪费——f数组是一个只有01两种值的数组，哪怕表示成bool类型都有点多余，于是我们可以考虑用bitset来优化它。

bitset你可以理解为一个长度很长的01数字串（实际上它是用int拼接而成），也可以理解为可以用位运算的bool数组

这样的话一行就可以一起求——如果我们用 f[i] 来表示第i行的01串：那么 f[i] = f [i] ( f[ i-1 ] << ( x[j]² ) )

然后bitset自带一个求1的个数的函数count，这就非常美滋滋了。

特别说明：bitset内部的实现其实是用int拼接形成所以时间复杂度是O（长度/32）的，那么本题的时间复杂度是O（n²/32）。

AC Code

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
bitset<N>a,b;
int main()
{
    int n;
    scanf("%d",&n);
    b[0]=1;
    for(int i=1;i<=n;i++)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        for(int j=l;j<=r;j++)   //能取的数是l到r
            a=a|b<<(j*j);
        b=a;
        a.reset();
    }
    printf("%d\n",b.count());
    system("pause");
    return 0;
}