NC17193 简单瞎搞题

题目链接

题目

题目描述

一共有 n个数，第 i 个数是 xi

xi 可以取 [li , ri] 中任意的一个值。

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

输入描述

第一行一个数 n。
然后 n 行，每行两个数表示 li,ri。

输出描述

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

示例1

输入

5
1 2
2 3
3 4
4 5
5 6

输出

26

备注

1 ≤ n , li , ri ≤ 100

题解

知识点：背包dp，STL。

多重背包思路，每个整数 \(x_i\) 当作 \([l_i,r_i]\) 个不加区分的 \(1\) 组成，再用一个数组 \(dp[i][j]\) 表示为考虑到 \(x_i\) 数 \(j\) 是否出现过。转移方程为：

\[dp[i][k] |= dp[i-1][k-j^2],j\in [l_i,r_i] \]

可以用 bitset 常数优化时间和空间。

时间复杂度 \(O(n(r-l) \cdot \frac{10^6}{32})\)

空间复杂度 \(O(n \cdot \frac{10^6}{32})\)

代码

#include <bits/stdc++.h>

using namespace std;

bitset<1000007> dp[107];

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    dp[0][0] = 1;
    for (int i = 1;i <= n;i++) {
        int l, r;
        cin >> l >> r;
        for (int j = l;j <= r;j++) {
            dp[i] |= dp[i - 1] << j * j;
        }
    }
    cout << dp[n].count() << '\n';
    return 0;
}