P6189 [NOI Online #1 入门组] 跑步 （DP/根号分治）

（才了解到根号分治这样的妙方法......）

将每个数当成一种物品，最终要凑成n，这就是一个完全背包问题，复杂度O(n²)，可以得80分（在考场上貌似足够了......）

#include <bits/stdc++.h>
//#define loveGsy
#define N 1000005
using namespace std;
int f[N];

int main() {
#ifdef loveGsy
    freopen("a.in", "r", stdin);
    freopen("a.out", "w", stdout);
#endif
    int n, p;
    cin >> n >> p;
    f[0] = 1;
    for (int i = 1; i <= n; i++)
        for (int j = i; j <= n; j++)
            f[j] = (f[j-i] + f[j]) % p;
    cout << f[n] <<endl; 
    return 0;
}

接着采用根号分治，将1~n这些数由sqrt(n)分成两半，前一半还是用完全背包的做法，复杂度O(n*sqrt(n))，再考虑另一种DP，gi,j表示从sqrt(n)~n之间选i个数，它们的和为j。

转移方程是g[i][j]=g[i][j-i]+g[i-1][j-m]。其中m就是sqrt(n)。g[i][j-i]可以理解为给数集中的每个数都加上1，g[i-1][j-m]理解为在数集中加入m这个数。两种DP统计完之后合并就行了（乘法原理），前一个为i，后一个就是n-i。

那么第二种DP为什么可以用呢？，因为从m~n之间选的数的个数一定不超过m，复杂度也就是O(n*sqrt(n))。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;

int n, p, m; ll ans = 0;
int f[N], g[410][N];

signed main() {
    cin >> n >> p;
    m = sqrt(n) + 1;
    f[0] = 1;
    for (int i = 1; i < m; i++)
        for (int j = i; j <= n; j++)
            f[j] = (f[j] + f[j - i]) % p;
    g[0][0] = 1;
    for (int i = 1; i < m; i++) 
        for (int j = i; j <= n; j++) {
            g[i][j] = g[i][j - i];
            if (j >= m) g[i][j] = (g[i][j] + g[i - 1][j - m]) % p;
        }
    for (int i = 0; i <= n; i++) {
        int sum = 0;
        for (int j = 0; j < m; j++) sum = (sum + g[j][n - i]) % p;
        ans = (ans + 1ll * f[i] * sum) % p;
    }
    cout << ans << endl;
    return 0;
} 

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。