Newcoder 132 C.简单瞎搞题（01背包+bitset）

Description

一共有$n$个数，第 $i$ 个数是 $x_i$ ，$x_i$ 可以取 $[l_i,r_i]$ 中任意的一个值。 设$S=\sum x_i^2$，求$S$种类数。

Input

第一行一个数$n$。
然后 $n$ 行，每行两个数表示 $l_i,r_i$。

$(1\le n,l_i,r_i\le 100)$

Output

输出一行一个数表示答案。

Sample Input

5
1 2
2 3
3 4
4 5
5 6

Sample Output

26

Solution

$01$背包，背包体积$100^3$，时间复杂度$O(100^5)$，用$bitset$优化一下复杂度为$O(\frac{100^5}{32})$

Code

#include<cstdio>
#include<bitset> 
using namespace std;
bitset<1000005>b[105];
int n,l[105],r[105];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d%d",&l[i],&r[i]);
	b[0].set(0);
	for(int i=1;i<=n;i++)
		for(int j=l[i];j<=r[i];j++)
			b[i]|=(b[i-1]<<(j*j));
	printf("%d\n",b[n].count());
	return 0;
}