BZOJ 2813: 奇妙的Fibonacci 线性筛

2813: 奇妙的Fibonacci

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 536  Solved: 152
[Submit][Status][Discuss]

Description

Fibonacci数列是这样一个数列：

F1 = 1， F2 = 1， F3 = 2 . . .

Fi = Fi-1 + Fi-2 (当 i >= 3)

pty忽然对这个古老的数列产生了浓厚的兴趣，他想知道：对于某一个Fibonacci数Fi，

有多少个Fj能够整除Fi (i可以等于j)，他还想知道所有j的平方之和是多少。

Input

第一行一个整数Q,表示Q个询问。

第二行四个整数:Q₁, A, B, C

第i个询问Q_i = (Q_i-1 * A + B) mod C + 1(当i >= 2)

Output

Ai代表第i个询问有多少个F_j能够整除F_Qi。

Bi代表第i个询问所有j的平方之和。

输出包括两行：

第一行是所有的Ai之和。

第二行是所有的Bi之和。

由于答案过大，只需要输出除以1000000007得到的余数即可。

Sample Input

2
2 2 1 8

Sample Output

6
55

HINT

对于100%的数据保证：Q <= 3*10^6，C <= 10^7，A <= 10^7，B <= 10^7，1 <= Q1<= C

证明有很多。。不看题解我不会

所以。。。

线性筛约数个数，平方和

#include<cmath>#include<ctime>#include<cstdio>#include<cstdlib>#include<cstring>#include<complex>#include<iostream>#include<algorithm>#include<iomanip>#include<vector>#include<string>#include<queue>#include<set>#include<map>using namespace std;typedef long long ll;inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return f*x;}const int N=10001000;const ll mod=1000000007;ll n,x,a,b,c;int prime[N],cnt,mn[N],cot[N],ci[N],sqr[N];//sqr Ô¼Êýƽ·½  cot Ô¼Êý¸öÊý  ci ×îСÖÊÒò×Ó´ÎÊý  mn  ³ýÈ¥×îСÖÊÒò×ÓµÄÔ¼Êý bool book[N];void initial(){sqr[1]=1;cot[1]=1;for(int i=2;i<=N;i++){if(!book[i]){prime[++cnt]=i;mn[i]=ci[i]=1;cot[i]=2;sqr[i]=(1ll*i*i+1)%mod;}for(int j=1;j<=cnt&&prime[j]*i<=N;j++){book[i*prime[j]]=1;mn[prime[j]*i]=i;ci[prime[j]*i]=1;cot[prime[j]*i]=cot[i]<<1;sqr[prime[j]*i]=(1ll*sqr[i]*prime[j]*prime[j]+sqr[i])%mod;if(i%prime[j]==0){mn[prime[j]*i]=mn[i];ci[prime[j]*i]=ci[i]+1;cot[prime[j]*i]=cot[i]/ci[prime[j]*i]*(ci[prime[j]*i]+1);sqr[prime[j]*i]=(1ll*sqr[i]*prime[j]*prime[j]+sqr[mn[i]])%mod;break;}}}}int main(){initial();n=read();x=read();a=read();b=read();c=read();a%=c;b%=c;int ans1=0,ans2=0;for(int i=1;i<=n;i++){if(i!=1)x=(x*a+b)%c+1;ans1+=(cot[x]+(x&1));ans2+=(sqr[x]+(x&1)*4);ans1%=mod;ans2%=mod;}printf("%d\n%d\n",ans1,ans2);return 0;}/*22  2  1  86 55*/