Codeforces Round #740 (Div. 2, based on VK Cup 2021 - Final (Engine)) D.Up the Strip

题目链接

题目大意：

有一个从 1到 n的单元格，你需要从 n走到 1，当你处于单元格 x(1<=x<=n)时，你有两种选择，

第一种选择为任意选择一个y（1<=y<x）到达x-y。

第二种选择为任意选择一个y（1<y<=x）到达x/y （向下取整）。

输出不同的方案数。

思路:

暴力O(n^2)时，很容易想到，第一种选择只需把前面各个位置的方案数加起来，前缀和即可。第二种选择我们就直接暴力加上每个的i/j。显然是会超时的。那么我们可以用数论分块来优化第二层的循环使得复杂度降低至O(n*sqrt(n))。这样子D1就通过了。

D2给的n为4e6,显然复杂度还需降低至O(n*log(n)).那么可以通过枚举倍数的方法。假设枚举i的倍数，那么2*i,3*i....n/i*i。[i*j,(i+1)*j-1] 这范围中的数除j时会等于i,那么我们可以通过差分对于cf[i*j]加上一个dp[i],对于cf[(i+1)*j]减去一个dp[i].这时候就很明了了。当处理到位置i时，只需将第一种方案的前缀合加上该位置差分的差分值就是答案了。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
ll dp[4000006];
ll cf[4000006];
int main(){
	ll n,m,sum=0;
	cin>>n>>m;
	dp[1]=1;//到达1的方案数有一种。
	for(ll i=1;i<=n;i++){
		cf[i]=(cf[i]+cf[i-1])%m;
		dp[i]=(dp[i]+sum+cf[i])%m;//不能写(dp[i]=sum+cf[i])%m 
		for(ll j=2;j*i<=n;j++){
			cf[j*i]=(cf[j*i]+dp[i])%m;
			cf[min(i*j+j,n+1)]=(cf[min(i*j+j,n+1)]+m-dp[i])%m;
		}
		sum=(sum+dp[i])%m;
	}
	cout<<dp[n]<<endl;
}