题目简述:
求下面这一坨东西的值:
⌊(b+d√2)n⌋mod7528443412579576937
题解:
我们先把里面玩意拿出来单独看
我们令
x1=b+d√2
x2=b−d√2
显然 x1,x2 是方程 x2−bx+b2−d4=0 的两根
考虑特征根方程,则我们可以令数列 {An} ,其中
An=(b+d√2)n+(b−d√2)n
那么 An 满足
An=bAn−1−b2−d4An−2
这东西可以用矩乘 O(lgn) 求,转移矩阵大概长这样:
⎡⎣⎢b1d−b240⎤⎦⎥
(转移的方法因人而异,看你用哪种)
然后要求的东西还要用 An 减去 (b−d√2)n 。这玩意显然是小于1大于等于0的。
所以在 An 的基础上再减1即可,当 b2=d 或 n <script type="math/tex" id="MathJax-Element-16">n</script>为奇数时不用减
P.S.刚开始交一直wa,然后wyf大爷一语道破:没考虑n=0的情况。。
#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ll;
typedef double db;
const int inf=0x3f3f3f3f;
int getint()
{
int f=1,g=0;char c=getchar();
while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0' && c<='9')g=(g<<3)+(g<<1)+c-'0',c=getchar();
return f*g;
}
const ll mod=7528443412579576937LL;
ll mul(ll a,ll b)
{
ll res=0;
for(;b;b>>=1,a=(a+a)%mod)
{
if(b&1)res=(a+res)%mod;
}
return res;
}
struct matrix{
ll a[3][3];
void init(){memset(a,0,sizeof a);}
void set(){a[1][1]=a[2][2]=1ll;}
matrix operator * (const matrix &b)const
{
matrix c;c.init();
for(int i=1;i<=2;i++)
{
for(int j=1;j<=2;j++)
{
for(int k=1;k<=2;k++)
{
c.a[i][k]=(c.a[i][k]+mul(a[i][j],b.a[j][k]))%mod;
}
}
}
return c;
}
matrix operator ^ (const ll &num)const
{
matrix res;res.init();res.set();
matrix a=*this;
for(ll i=num;i;i>>=1,a=a*a)
{
if(i&1)res=res*a;
}
return res;
}
};
ll b,d;
ll n;
int main()
{
// freopen("in.txt","r",stdin);
cin>>b>>d>>n;
if(n==0)
{
puts("1");
return 0;
}
matrix trans;trans.init();
matrix now;now.init();
now.a[1][1]=b;
now.a[2][1]=2;
trans.a[1][1]=b;
trans.a[1][2]=(d-b*b)>>2;
trans.a[2][1]=1ll;
trans=trans^(n-1);
now=trans*now;
ll ans=now.a[1][1];
ans-=(d!=b*b&&~n&1);
printf("%lld\n",ans);
return 0;
}