Codeforces Round #722 (Div. 1)_B（线性dp）

Codeforces Round #722 (Div. 1)_B

题目大意：

$2n$个在$x$轴，第$i$个点所在的位置为$x=i$。
从$2n$个点选出$n$个区间（每个点都要被选），如果满足任意两个区间满足以下任意一种情况：

• 大的区间完全包含小区间
• 两个区间有相同长度

则说明是好的区间序列，问：有多少个好的区间序列，答案对998244353取模。

解题思路：

设$dp[i]$，为$2i$个点的总方案。

引理：定义x为匹配1的点。当$p(x<p<=2n)$属于线段区间某个端点，则它的长度肯定等于$[1,x]$的长度

• $x>n$：那么肯定有个区间为$[2n-x+1,2n]$，而剩下的区间都处在$[2n-x+2,x-1]$, 而区间长度为$2x-2n-2$，那么对应的dp值就是$d{p}_{x-n-1}$
• $x<=n$：长度肯定都相等且都是$n$的约数，才能填满$2n$个点，$n$个区间

式子:$dp[i]={\sum }_{j=0}^{i-1}dp[j]+D(i)（i的因子个数）-1$

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
const int mod = 998244353;
bool vis[maxn];
int cnt, n, p[maxn], dp[maxn];
void Euler_pri() {
    vis[0] = vis[1] = true;
    for (int i = 2; i <= n; i++) {
        if (!vis[i]) p[++cnt] = i;
        for (int j = 1; i * p[j] <= n; j++) {
            vis[i * p[j]] = true;
            if (i % p[j] == 0) break;
        }
    }
}
int resl(int x) {
    int res = 1;
    for (int i = 1, tmp = 0; i <= cnt; i++) {
        if (!vis[x]) {
            res = res * 2 % mod;
            break;
        }
        if (x % p[i]) continue;
        tmp = 0;
        while (x % p[i] == 0) x /= p[i], tmp++;
        res = res * (tmp + 1) % mod;
        if (x == 1) break;
    }
    return res;
}
int sum[maxn];
int main() {
    cin >> n;
    Euler_pri();
    dp[0] = sum[0] = 1;
    for (int i = 1; i <= n; i++) {
        dp[i] = (sum[i - 1] + resl(i) - 1) % mod;
        sum[i] = (sum[i - 1] + dp[i]) % mod;
    }
    cout << dp[n] << endl;
}