2018 ICPC 沈阳站 M. Renaissance Past in Nancy 题解

M. Renaissance Past in Nancy

题目链接
题意：
$T\le 10$。有$n\le 10^4$种物品，每种物品数量和体积分别为$1\le a,b\le 10^3$，$m\le 10^4$询问，每次询问区间$1\le l,r\le n$的物品装体积为$c\le 10^3$的方案数。
可逆背包：
对于背包问题，可以逆向转移而达到移除一个物品的贡献。
计数背包可逆，而求max背包不可逆。
背包与生成函数：
01背包：$F(x)=1+x^b$
01退背包：$F(x)=1-x^b$
完全背包：$F(x)=\frac{1}{1-x^b}$
多重背包：$F(x)=\frac{1-x^{(a+1)b}}{1-x^b}$
一堆物品生成函数：$G(x)={\prod }_{所有物品}F(x)$
一堆物品，背包体积为$c$的方案数${\sum }_{i=0}^v{g}_i=(\frac{G(x)}{1-x})[c]$
若$H(x)={\sum }_{i=0}{h}_i{x}^i$，$\frac{H(x)}{1-x}={\sum }_{i=0}{\sum }_{j=0}^i{h}_j{x}^i$。
本题：
记第$i$个物品的生成函数为$F_i(x)$，$G_i(x)={\prod }_{j=1}^i{F}_j(x)$。
特别的$G_0(x)=1$。
那么答案等于$(\frac{1}{1-x}{G}_r(x)\ast \frac{1}{G_{l-1}(x)})[c]$。
$G_i(x)$的求法：
从生成函数的角度，多重背包等价于体积为$b$的完全背包和体积为$(a+1)b$的01退背包组成。
用背包DP的方法维护$G_i(x)$，$O(c)$添加一个物品。
$\frac{1}{G_i(x)}$的求法：
类似的，多重背包的逆等价于体积为$(a+1)b$的完全背包和体积为$b$的01退背包组成。
关于询问：
每次询问求一次卷积再求前缀和，复杂度太高。
预处理出$\frac{1}{1-x}\frac{1}{G_i(x)}$或$\frac{1}{1-x}{G}_i(x)$,（就是求前缀和啦），$O(nc)$；每次询问就只用求$(\frac{1}{1-x}{G}_r(x)\ast \frac{1}{G_{l-1}(x)})$的第$c$项，$O(c)$解决。
最后，感谢这篇博客给的提示。
Code：

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;

void insert(vector<int> &f, int a) {
    for (int i = a; i <= f.size() - 1; i++) {
        f[i] = (f[i] + f[i - a]) % mod;
    }
}

void remove(vector<int> &f, int a) {
    for (int i = f.size() - 1; i >= a; i--) {
        f[i] = (f[i] - f[i - a] + mod) % mod;
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int _;
    cin >> _;
    while (_--) {
        int n, m;
        cin >> n >> m;
        const int V = 1000;
        vector<vector<int>> f(n + 1, vector<int>(V + 1)), g(f);
        f[0][0] = g[0][0] = 1;
        for (int i = 1; i <= n; i++) {
            int a, b;
            cin >> b >> a;//b个a
            f[i] = f[i - 1];
            g[i] = g[i - 1];
            insert(f[i], a);
            remove(f[i], a * (b + 1));
            remove(g[i], a);
            insert(g[i], a * (b + 1));
        }
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= V; j++)
                f[i][j] = (f[i][j] + f[i][j - 1]) % mod;
        }
        int last = 0;
        static int Case = 0;
        cout << "Case #" << ++Case << ":\n";
        while (m--) {
            int l, r, c;
            cin >> l >> r >> c;
            int t1 = (l + last) % n + 1;
            int t2 = (r + last) % n + 1;
            l = min(t1, t2), r = max(t1, t2);
            int res = 0;
            for (int i = 0; i <= c; i++)
                res = (res + (ll) g[l - 1][i] * f[r][c - i]) % mod;
            last = res;
            cout << res << "\n";
        }
    }
}