dp--bitset优化

Problem Description

一共有\(n\)个数，第\(i\)个数是\(x_i\)，\(x_i\)可以取\([l_i , r_i]\)中任意的一个值。

设\(S = \sum {x_i}^2\)，求\(S\)种类数。

Analysis of ideas

C++的 bitset 在 bitset 头文件中，它是一种类似数组的结构，它的每一个元素只能是０或１，每个元素仅用１bit空间。

我们可以利用 bitset 的每一位对应可以取到的数字

比如说 10110000 代表可以取 0,2,3 ,此时来一个数４，那么对应可以取的数就是 00001011 ,也就是 4,6,7

Accepted code

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define pii pair<int,int>
const int inf = 0x3f3f3f3f;
const int maxn = 201110;
const int M = 1e9+7;
int n,m,k,ok;
const int N = 105;
bitset<N*N*N> a, b; 

signed main()
{
    cin>>n;
    b[0] = 1;
    for(int i = 1,l,r; i <= n; i++) 
    {
        cin>>l>>r;
        for(int j = l; j <= r; j++)
        {
            a |= b << j*j;
        }
        b = a;
        a.reset();
    }
    cout<<b.count()<<endl;
    return 0;
}