题目链接
dp+刷表
考虑用除的方法到达i位置,则 f [ i ] f[i] f[i] 对 [ i × j , i × j + i − 1 ] [i\times j,\ i\times j+i-1] [i×j, i×j+i−1] 都有贡献,所以可以枚举倍数,像埃氏筛一样刷。区间修改用树状数组会T,注意到查询位置单增,修改的位置均在最近的查询位置之后,所以用差分维护。总复杂度 O ( n l o g n ) O(nlogn) O(nlogn)
完整代码:
#include<bits/stdc++.h>
#define FAST ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define INF 0x3f3f3f3f
typedef long long ll;
const int maxn = 4e6+5;
using namespace std;
ll n,p;
ll c[maxn],f[maxn],sum[maxn];
void add(int l, int r, ll val)
{
c[l]=(c[l]%p+val%p)%p;
c[r+1]=(c[r+1]%p-val%p+p)%p;
}
int main()
{
FAST;
while(cin>>n>>p)
{
memset(c,0,sizeof(c));
ll sum=0,sum1=0;
sum=f[1]=1;
for (ll i=2; i<=n; i++) add(i,min(2*i-1,n),f[1]);
for (ll i=2; i<=n; i++)
{
sum1=(sum1%p+c[i]%p)%p;
f[i]=(sum%p+sum1%p)%p;
sum=(sum%p+f[i]%p)%p;
for (ll j=1; j<=n/i; j++)
add(j*i+(j==1),min(j*i+j-1,n),f[i]%p);
}
cout<<f[n]%p<<endl;
}
return 0;
}