【牛客contest16806 C】【状态压缩】【bitset】

题目链接

题目描述
一共有 n个数，第 i 个数是 xi
xi 可以取 [li , ri] 中任意的一个值。
设 $S=\sum {x_i}^2$，求 S 种类数。

n, l, r 都是1~100

首先考虑暴力做法，因为n很小，暴力枚举是10^6
用dp[100][10^6] 这样一个数组来存结果。第一维i表示用了i个数，第二维j表示要凑出j这个数，dp[i[][j]取值只有0和1，表示这个数能不能凑出来。

首先bool数组可以用bitset来节省空间，bitset每位数只占1bit，是bool的八分之一。

接下来我们考虑第一维是可以被优化掉的，因为我们不需要中间用了i个数的结果，最后我们是n个数都要用上的。

我们首先考虑只用两个数组 mark[10^6]和mark1[10^6]
mark1[k + j * j] |= mark[k]
一遍循环之后再令mark[k] = mark1[k]
由于mark1数组是一个待更新的全0状态，我们不难发现从mark到mark1其实就是左移了j*j位

我们直接把这个bool型的mark数组看成一个01串，也就是bitset<10^6> a[100]，即长为10^6的这样100个01串，用来存取i个数时所能凑成哪些数。开100个就不用像mark和mark1那样来回赋值了。

最后用a[n].count()直接输出串中1的个数