[题解] CF 1528B Kavi on Pairing Duty

[题解] CF 1528B Kavi on Pairing Duty 动态规划+简单数论

题目链接

这是题目给出的部分样例。
让我们仔细看第$2$个和第$7$个。好像第$7$个内层和第$2$个完全一样?
这让我们反应过来，好像可以$dp$。
我们经过进一步的推理可以发现:
设第$1$个点和第$x$个点相连，那么对于有端点在$[x+1,2n]$的线段，
我们发现，如果这个线段的左端点在$[1,x]$之内，那么这两个线段必相交，故长度一定相等;
如果这个线段的左端点不在$[1,x]$内部，那么他们两个的长度还是必须相等。
有了这个结论，我们就能顺利的推式子了。
设$dp[i]$为由$2i$个点时总的方案数，那么我们考察与第$1$个节点相连的边的长度$len$：(这里我们仍然应用上个结论中的点$x$)
$Case1:len\ge n+1$，那么对于在$x$右边的点显然左端点只能在$[1,x]$以内。把所有的在$[x+1,2n]$的点操作完之后，我们发现中线还胜了一部分点没有配对，这时候只需要利用之前的状态就行了。
再通过观察样例，可以知道所有$len\in [n+1,2n-1]$产生的贡献为$$\sum _{i=1}^{n-1}dp[i]$$
显然需要用前缀和优化。
$Case1:len\le n$，那么根据我们之前的结论和观察样例，所有的线段长度一定一样，不管是和第$1$个相交的还是不相交的。
我们在仔细观察样例，可以发现左端点在$[1,x-1]$的线段构成一个小块，长度为$2\ast len$。如果想要为$dp[i]$做贡献，必有$2\ast len|2\ast n$，即$$len|n$$故此时能够产生的总贡献为$d(n)$.
综上所述，可以推出来这个式子:$$dp[n]=\sum _{i=1}^{n-1}dp[i]+d(n)$$
最后用样例检查了一下这个式子，发现没问题。
用线性筛筛出$d$，时间复杂度$O(n)$。

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
using namespace std;

const double eps = 1e-10;
const double pi = acos(-1.0);
const int maxn = 1e6 + 10;
const ll mod = 998244353;

int n;
ll dp[maxn];
ll sum[maxn];

int p[maxn],cnt;
int a[maxn];//i的最小素因子次数
int d[maxn];//约数个数
bool isp[maxn];

void getp(){
	isp[1] = 1;
	d[1] = 1;
	for(int i = 2; i < maxn; i++){
		if(!isp[i]){
			p[++cnt] = i;
			a[i] = 1;
			d[i] = 2;
		}
		for(int j = 1; j <= cnt && i*p[j] < maxn; j++){
			isp[i*p[j]] = 1;
			if(i%p[j] == 0){
				a[i*p[j]] = a[i]+1;
				d[i*p[j]] = d[i]/(a[i]+1)*(a[i]+2);
				break;
			}
			else{
				a[i*p[j]] = 1;
				d[i*p[j]] = 2*d[i];
			}
		}
	}
}

void solve(){
	getp();
	scanf("%d",&n);
	dp[1] = sum[1] = 1;
	for(int i = 2; i <= n; i++){
		dp[i] = (sum[i-1]+d[i]) % mod;
		sum[i] = (sum[i-1]+dp[i]) % mod;
	}
	printf("%lld\n",dp[n]);
}

int main()
{
	solve();
	return 0;
}