Codeforces Round #740 D1. Up the Strip (整数分块 + 前缀和)

题目链接

题意

你有两个个整数$n（2\le n\le 2\ast 10^5）,m(10^8<m<10^9)$，$m$为素数

可以进行两种操作

• 选择一个数$y(1\le y<x)$，将$x$ 变为$x-y$
• 选择一个数$z(2\le z\le x)$，将$x$变为$x/z$

问有多少种方案数可以将整数$n$变为$1$，输出方案数对$m$取模

思路

考虑$dp$，$dp[i]$表示整数$i$变为$1$的方案数

容易得出转移方程为 $dp[i]={\sum }_{j=1}^{i-1}dp[j]+{\sum }_{j=2}^{i}dp[i/j]$

如果直接求解时间复杂度为$O(n^2)$，则需要优化

容易看出${\sum }_{j=1}^{i-1}dp[j]$是一个前缀和，用一个变量$sum$存下即可

${\sum }_{j=2}^{i}dp[i/j]$ 中的$i/j(2\le j\le i)$ 容易看出是整数分块

tips

下面是整数分块的解释

对于一个整数$n$，容易知道存在一个区间$\forall i\in [l,r]$ 有$n/i$相等

整数分块就是告诉你如果已知$l$，那么$r=n/(n/l)$，至于为什么？这里以后有时间了再补吧

for(int l = 1,r ; l <= n ; l = r + 1){
    r = n / (n / l);//区间[l,r]内 n/i的值都为n/l
}

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read() {
    ll s = 0, w = 1;
    char ch = getchar();
    while (ch < 48 || ch > 57) {
        if (ch == '-') w = -1;
        ch = getchar();
    }
    while (ch >= 48 && ch <= 57)
        s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
    return s * w;
}
ll f[200005];
int main() {
    f[1] = 1;
    ll n = read(), mod = read();
    ll sum = 0;//sum为前缀和
    for (int i = 2; i <= n; ++i) {
        sum = (sum + f[i - 1]) % mod;
        f[i] = sum;
        for (int l = 2, r; l <= i; l = r + 1) {//因为n/i时i是从2开始
            r = i / (i / l);
            f[i] = (f[i] + (r - l + 1) * f[i / l]) % mod;
        }
    }
    cout << f[n] << endl;
}