51nod 1597 有限背包计数问题

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5, mod = 23333333;
int f[2][maxn];//用了前i个数[1, sqn)，和为j的方案数，i维滚动
int g[2][maxn];//用了i个数，和为j的方案数。放一个sqn或者全部都加1
int h[maxn];//只用大于等于sqn的数，和为i的方案数
int pre[maxn];//pre[v]: j % i = v的f值之和
int main()
{
    int n;
    cin >> n;
    int sqn = sqrt(n) + 1;

    //f[i][j]=∑(k=0...i)f[i−1][j−k∗i]
    //(j-k*i)%i=j%i
    //所以得用pre[v]保存上一次的j%i=v的所有f[i-1][j]的值，但要求j>=i*i，因为i最多i个，j也不能超过这次的j。
    int now = 1;
    f[1][0] = f[1][1] = 1;
    for (int i = 2; i < sqn; ++i){
        now ^= 1;
        //j < i的时候照搬就好了，所以pre[j]为0
        for (int j = 0; j < i; ++j) pre[j] = 0;
        int m = -1;//减少取模次数
        for (int j = 0; j <= n; ++j){
            ++m;//模拟取模
            if (m >= i) m = 0;
            //把上一次的值加上
            //此时pre[m]已经保存了上次的所有j之前的j%i=m的f[i-1][j]的和了
            pre[m] += f[now ^ 1][j];
            pre[m] %= mod;
            f[now][j] = pre[m];//就是这次的结果了
            //i最多i个，所以每次得把往前数第i个减掉
            if (j - i * i >= 0) pre[m] -= f[now ^ 1][j - i * i];
            if (pre[m] < 0) pre[m] += mod;
        }
    }

    g[0][0] = 1;//这玩意会影响滚动，所以单独转移一遍然后置零
    now = 1;
    for (int j = sqn; j <= n; ++j){
        g[now][j] = (g[now][j - 1] + g[now ^ 1][j - sqn]) % mod;
        h[j] = (h[j] + g[now][j]) % mod;
    }
    g[0][0] = 0;
    for (int i = 2; i < sqn; ++i){
        now ^= 1;
        for (int j = sqn; j <= n; ++j){
            g[now][j] = 0;
            g[now][j] = (g[now][j - i] + g[now ^ 1][j - sqn]) % mod;
            h[j] = (h[j] + g[now][j]) % mod;
        }
    }
    h[0] = 1;
    int ans = 0;
    for (int i = 0; i <= n; ++i){
        ans += (ll)f[now][i] * h[n - i] % mod;
        ans %= mod;
    }
    cout << ans << endl;
    return 0;
}