本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作。
本文作者:ljh2000
作者博客:http://www.cnblogs.com/ljh2000-jump/
转载请注明出处,侵权必究,保留最终解释权!
题目链接:BZOJ4818
正解:卷积+快速幂
解题报告:
很simple的一道题,每次转移就是一个卷积,用快速幂优化这个过程。
对于至少有一个质数的限制,用容斥减掉没有质数的情况就好了。
//It is made by ljh2000
//有志者,事竟成,破釜沉舟,百二秦关终属楚;苦心人,天不负,卧薪尝胆,三千越甲可吞吴。
#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
#include <string>
#include <queue>
#include <cmath>
#include <ctime>
#define lc root<<1
#define rc root<<1|1
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define reg(i,x) for(int i=first[x];i;i=next[i])
using namespace std;
typedef long long LL;
const int mod = 20170408;
const int MAXN = 20000011;
int n,m,p,prime[MAXN],cnt;
bool vis[MAXN];
LL ans,a[150],b[150],c[150];
inline int getint(){
int w=0,q=0; char c=getchar(); while((c<'0'||c>'9') && c!='-') c=getchar();
if(c=='-') q=1,c=getchar(); while (c>='0'&&c<='9') w=w*10+c-'0',c=getchar(); return q?-w:w;
}
inline void init(){
vis[1]=1;
for(int i=2;i<=m;i++) {
if(!vis[i]) { prime[++cnt]=i; }
for(int j=1;j<=cnt && prime[j]*i<=m;j++) {
vis[i*prime[j]]=1;
if(i%prime[j]==0) break;
}
}
}
inline void mul(LL *a,LL *b){
memset(c,0,sizeof(c)); int now;
for(int i=0;i<p;i++)
for(int j=0;j<p;j++) {
now=(i+j)%p;
c[now]+=a[i]*b[j];
c[now]%=mod;
}
}
inline LL solve(){
int y=n; memset(b,0,sizeof(b));
b[0]=1;
while(y>0) {
if(y&1) {
mul(b,a);
for(int i=0;i<p;i++) b[i]=c[i];
}
mul(a,a);
for(int i=0;i<p;i++) a[i]=c[i];
y>>=1;
}
return b[0];
}
inline void work(){
n=getint(); m=getint(); p=getint();
init();
for(int i=1;i<=m;i++) a[i%p]++;
ans=solve();
memset(a,0,sizeof(a));
for(int i=1;i<=m;i++) if(vis[i]) a[i%p]++;
ans-=solve();
ans+=mod; ans%=mod;
printf("%lld",ans);
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("count.in","r",stdin);
freopen("count.out","w",stdout);
#endif
work();
return 0;
}
//有志者,事竟成,破釜沉舟,百二秦关终属楚;苦心人,天不负,卧薪尝胆,三千越甲可吞吴。