luogu P3263 [JLOI2015]有意义的字符串

luogu

兄弟会背叛你，女人会离开你，金钱会诱惑你，生活会刁难你，只有数学不会，不会就是不会，怎么学都不会。

先记(A=frac{b+sqrt{d}}{2}),出现这种东西不妨考虑他的"共轭项"",记(B=frac{b-sqrt{d}}{2}).可以发现(A+B=b,AB=frac{b^2-d}{4}),并且由于(bmod 2=1,dmod 4=1),所以(AB)的值一定是个整数

现在要求(lfloor A^n floor),由于我们有(A+B,AB),所以可以考虑求(lfloor A^n+B^n-B^n floor),记(f(i)=A^i+B^i),有(f(i)=(A+B)f(i-1)+ABf(i-2)),所以(f(i))可以矩阵快速幂求,后面的(-B^n),有(-1<-B^n<1),那只要讨论一下(B)是否为0以及(n)的奇偶性即可

#include<bits/stdc++.h>
#define LL long long
#define uLL unsigned long long

using namespace std;
const uLL mod=7528443412579576937;
uLL rd()
{
    uLL x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+(ch^48);ch=getchar();}
    return x*w;
}
uLL pls(uLL a,uLL b){a+=b,a-=a>=mod?mod:0;return a;}
uLL mul(uLL a,uLL b){uLL an=0;while(b){if(b&1) an=pls(an,a);a=pls(a,a),b>>=1;}return an;}
struct matrix
{
    uLL a[2][2];
    matrix(){memset(a,0,sizeof(a));}
    matrix operator * (const matrix &bb) const
    {
    	matrix an;
    	for(int i=0;i<=1;++i)
    	    for(int j=0;j<=1;++j)
    		    an.a[i][j]=pls(mul(a[i][0],bb.a[0][j]),mul(a[i][1],bb.a[1][j]));
    	return an;
    }
}aa,bb;
uLL n,b,d;

int main()
{
    b=rd(),d=rd(),n=rd();
    if(!n){puts("1");return 0;}
    bool fg=(n&1)^1,f2=b*b==d;
    aa.a[0][0]=b%mod,aa.a[0][1]=f2?0:2;
    bb.a[0][0]=b%mod,bb.a[1][0]=pls(0,mod-((LL)b*(LL)b-(LL)d)/4),bb.a[0][1]=1;
    --n;
    while(n)
    {
    	if(n&1) aa=aa*bb;
    	bb=bb*bb,n>>=1;
    }
    printf("%llu
",pls(aa.a[0][0],mod-(!f2&&fg)));
    return 0;
}