bzoj 4833: [Lydsy1704月赛]最小公倍佩尔数 数论

题意

令$(1+\sqrt 2)^n=e(n)+f(n)*\sqrt 2$,其中$e(n),f(n)$都是整数，显然有$(1-\sqrt2)^n=e(n)-f(n)*\sqrt2$。令g(n)表示f(1),f(2)…f(n)的最小公倍数，给定两个正整数n和p,其中p是质数，并且保证f(1),f(2)…f(n)在模p意义下均不为0,请计算sigma(i*g(i)),1<=i<=n.其在模p的值。
$1≤n≤10^6,2≤p≤10^9+7$

分析

首先要观察出$f(0)=0,f(1)=1,f(n)=2f(n-1)+f(n-2)$
然后再大力猜一波结论，有

$$gcd(f(i),f(j))=f(gcd(i,j))$$
然后剩下的就跟 这题一样了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=1000005;

int n,MOD,f[N],h[N];

int ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

int main()
{
    int T;scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&n,&MOD);
        f[0]=0;f[1]=h[1]=1;
        for (int i=2;i<=n;i++) f[i]=(f[i-1]*2%MOD+f[i-2])%MOD,h[i]=f[i];
        for (int i=1;i<=n;i++)
        {
            int ny=ksm(h[i],MOD-2);
            for (int j=i*2;j<=n;j+=i) h[j]=(LL)h[j]*ny%MOD;
        }
        int ans=0,s=1;
        for (int i=1;i<=n;i++) s=(LL)s*h[i]%MOD,(ans+=(LL)s*i%MOD)%=MOD;
        printf("%d\n",ans);
    }
    return 0;
}