链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题目描述
一共有 n个数,第 i 个数是 xi
xi 可以取 [li , ri] 中任意的一个值。
设 S=∑xi2S = \sum{{x_i}^2}S=∑xi2,求 S 种类数。
输入描述:
第一行一个数 n。 然后 n 行,每行两个数表示 li,ri。
输出描述:
输出一行一个数表示答案。
示例1
输入
5 1 2 2 3 3 4 4 5 5 6
5 1 2 2 3 3 4 4 5 5 6
输出
26
26
备注:
1 ≤ n , li , ri ≤ 100
题目说是简单瞎搞题,实际上如果方法没找到还是有不小难度。开始看完题目第一反应这不就是背包问题,一写果然不出意料超时了,翻看大佬写法领悟到这一题可以用bitset和位运算来优化.
设区间(x,y),每一轮有y-x+1种取法,每次只能取一种,同组相互之间不干扰,状态只和上一次有关。假设一个格子z已经有物品,要放一个t体积的物品,把这个物品其实就是把这个格子z清空,把z+t填满,可以形象理解为把这个格子往左移了t格,自然想到位运算。当前的组开始清空,把上一轮装满的格子全部左移t个赋值给当前状态,再计算当前的格子有多少个装满,也就是有几个1就是所有方法数,不要忘记最开始的方法数是1.
代码:
#include<stdio.h>
#include<algorithm>
using namespace std;
#define r(i,a,b)for(int i=a;i<=b;i++)//定义一个循环
bitset<1000001>m[110];//bitset不宜开过大,否则移动耗时久容易超时
int main()
{
int n; scanf("%d", &n);
int x, y;//范围
m[0] = 1;//初始位置
r(i, 1, n)
{
scanf("%d%d", &x, &y);
r(j, x, y)
m[i] |= m[i - 1] << j * j;//把上一次的数整体移动,这一次的实际整排都是0,有1的情况
}
printf("%d\n", m[n].count());
return 0;
}