【BZOJ2813】奇妙的Fibonacci

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个询问。
第二行四个整数:Q1, A, B, C
第i个询问Qi = (Qi-1 * A + B) mod C + 1(当i >= 2)
Output

Ai代表第i个询问有多少个Fj能够整除FQi。
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

Source

某人出的胡策题里的唯一一个送分题.
随便感受一下,fib能不能模肯定和项数有关
然后打个表发现,fib_i是所有项数是i的因子的数的整数倍
所以答案就是约数个数和一个平方和
然后随便线筛一下,就求出来了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define P 1000000007
#define MAXN 10000010
using namespace std;
int T,n,a,b,c;
bool not_prime[MAXN];
int prime[MAXN],top;
int sqr[MAXN],cnt[MAXN],minn[MAXN],d[MAXN];
//sqr 约数平方  cnt 约数个数   minn 最小质因子次数  d  除去最小质因子的约数 
int ans1,ans2;
void check()
{
    sqr[1]=1;cnt[1]=1;
    for (int i=2;i<=c;i++)
    {
        if (!not_prime[i])  cnt[i]=2,sqr[i]=(1ll*i*i+1)%P,prime[++top]=i,minn[i]=d[i]=1;
        //cnt - g sqr -  f d - d minn - e
        for (int j=1;j<=top&&i*prime[j]<=c;j++)
        {
            not_prime[i*prime[j]]=1;minn[i*prime[j]]=1;
            d[i*prime[j]]=i;cnt[i*prime[j]]=cnt[i]<<1;
            sqr[i*prime[j]]=1ll*sqr[i]*((1ll*prime[j]*prime[j]+1)%P)%P;
            if (i%prime[j]==0)
            {
                minn[i*prime[j]]=minn[i]+1;d[i*prime[j]]=d[i];
                cnt[i*prime[j]]=(cnt[i]/(minn[i]+1))*(minn[i*prime[j]]+1);
                sqr[i*prime[j]]=(1ll*sqr[i]*(1ll*prime[j]*prime[j]%P)+sqr[d[i]])%P;
                break;
            }
        }
    }
}
int main()
{
    cin>>T>>n>>a>>b>>c;a%=c;b%=c;
    check();
    while (T--)
    {
        ans1+=cnt[n]+(n&1);ans2+=sqr[n]+(n&1)*4;
        if (ans1>=P)    ans1%=P;
        if (ans2>=P)    ans2%=P;
        n=(1ll*n*a+b)%c+1;
    }
    cout<<ans1<<endl<<ans2<<endl;
}