题目大意
是否存在整数m使得 (1+√2)n=√m+√(m−1)
结论
首先n=1存在解。
假设n=k时存在解,易证n=k+1也必然存在解。
于是设
(1+√2)n=an+bn√2
a与b可以用矩阵乘法快速算出,再分类讨论n的奇偶性得到m。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int mo=1000000007;
struct dong{
int a[3][3];
friend dong operator *(dong a,dong b){
int i,j,k;
dong c;
fo(i,1,2)
fo(j,1,2)
c.a[i][j]=0;
fo(k,1,2)
fo(i,1,2)
fo(j,1,2)
c.a[i][j]=(c.a[i][j]+(ll)a.a[i][k]*b.a[k][j]%mo)%mo;
return c;
}
};
dong kqp,ans,tmp;
int i,j,k,l,t,m,top,s[80];
ll n,nn,a,b;
int main(){
scanf("%lld",&n);
nn=n;
if (n==1){
printf("2\n");
return 0;
}
ans.a[1][1]=ans.a[2][2]=1;
ans.a[2][1]=ans.a[1][2]=0;
tmp.a[1][1]=tmp.a[1][2]=tmp.a[2][2]=1;
tmp.a[2][1]=2;
n--;
while (n){
s[++top]=n%2;
n/=2;
}
while (top){
ans=ans*ans;
if (s[top--]) ans=ans*tmp;
}
kqp.a[1][1]=kqp.a[1][2]=1;
ans=kqp*ans;
a=ans.a[1][1];b=ans.a[1][2];
if (nn%2) m=(ll)2*b%mo*b%mo;else m=(ll)a*a%mo;
printf("%d\n",m);
}