codeforces 1561D Up the Strip

D. Up the Strip

题意： 目前是x，有两种操作：1. $x\to x-y$。 2. $x\to \lfloor \frac{x}{z}\rfloor$。问从n到1的方案数。

根据定义得： $f(x)={\sum }_{y=1}^{x-1}f(x-y)+{\sum }_{z-2}^{x}f(\lfloor \frac{x}{z}\rfloor )$

D1的subtask n是2e5， 刚开始以为按cf的尿性大概nlogn，后来摸了下才感觉到是整数分块，根号，还是蔡了。

前半段前缀和维护，后半段整数分块。$z<\sqrt{n}$ 按z枚举，$z\ge \sqrt{n}$ 按结果枚举，对每个结果定左右边界

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll N = 2e5 + 10;

ll n, mod;
ll dp[N], pre[N];

ll getAns(ll x) {
    ll res = 0;
    for (ll i = 1; i <= x / i; ++i) {
        ll r = x / i;
        ll l = x / (i + 1);
        ll tmp = (((r - l) % mod )* (dp[i] % mod) )% mod;
        res = (res + tmp) % mod;
    }
    for(ll i = 1; i < x/ i; ++ i) {
        res = (res+ dp[x / i])%mod;
    }
    return res;
}

signed main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);

    cin >> n >> mod;

    dp[0] = dp[1] = 1;
    pre[1] = 1;
    for (ll i = 2; i <= n; ++i) {
        dp[i] = (pre[i - 1] + getAns(i)) % mod;
        pre[i] = (pre[i - 1] + dp[i]) % mod;
    }
    cout << dp[n] << endl;
    return 0;
}

D2 4e6，代码最后其实比D1短。

考虑角度是：从f(x + 1) 比 f(x) 多了什么入手。

把式子重新搬过来：

$f(x)={\sum }_{y=1}^{x-1}f(x-y)+{\sum }_{z-2}^{x}f(\lfloor \frac{x}{z}\rfloor )$

前半段多了一个f(x)；

后半段多了一个f(1)， （因为 $\frac{x+1}{x+1}=1$），另外每个分子x都变成了x+1，但因为下取整，只有在z|x的时候有影响，反过来用类似埃氏筛的思想对每个$z|x,\lfloor \frac{x}{z}\rfloor =c$ 的c去枚举，并对c的每个倍数进行$+f(c)-f(c-1)$，

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll N = 4e6 + 10;

ll n, mod;
ll dp[N];


signed main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);

    cin >> n >> mod;

    dp[0] = dp[1] = 1;
    dp[2] = 2;
    for (ll i = 4; i <= n; i += 2) dp[i] = (dp[i] + 1) % mod;
    for (ll i = 3; i <= n; ++i) {
        dp[i] = (dp[i] + 2 * dp[i - 1] + dp[1]) % mod;
        ll sub = dp[i] - dp[i - 1];
        for (ll j = 2 * i; j <= n; j += i) {
            dp[j] = (dp[j] + sub + mod) % mod;
        }
    }

    cout << dp[n];
    return 0;
}