该博客探讨了一种优化算法，用于解决在限制范围内的数列平方和的计数问题。通过位运算和动态规划，将原本的二维数组优化为一维的bitset，大大减少了空间复杂度。博主详细介绍了如何利用位运算技巧，尤其是通过左移操作，实现动态规划的状态转移，并最终计算出结果。这种方法适用于小规模输入，如n、l和r都在1到100之间的情况。
题目描述
一共有 n个数，第 i 个数是 xi
xi 可以取 [li , ri] 中任意的一个值。
设
S
=
∑
x
i
2
S = \sum{{x_i}^2}
S=∑xi2，求 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的个数
扫一扫
1831
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
