51NOD 1537 分解

1537 分解
基准时间限制：0.5 秒 空间限制：131072 KB 分值: 80 难度：5级算法题

问(1+sqrt(2)) ^n 能否分解成 sqrt(m) +sqrt(m-1)的形式
如果可以 输出 m%1e9+7 否则 输出no

Input

一行，一个数n。（n<=10^18）

Output

一行，如果不存在m输出no，否则输出m%1e9+7

Input示例

2

Output示例

9

---

找奇偶性规律+快速幂
(1+sqrt(2) )^n 可以表示为x+y*sqrt(2)
打表发现

n为偶数时 m=x*x
n为奇数时 m=x*x+1

显然 xn,yn满足

xn   =  1    2   *  x(n-1)
yn      1    1      y(n-1)

矩阵快速幂

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>

void mult(ll res[2][2],ll x[2][2]){
    ll t[2][2];
    for(int i=0;i<2;++i)
        for(int j=0;j<2;++j){
            t[i][j]=0;
            for(int k=0;k<2;++k)
                t[i][j]=(t[i][j]+res[i][k]*x[k][j])%INF;
        }
    for(int i=0;i<2;++i)
        for(int j=0;j<2;++j)
            res[i][j]=t[i][j];
}

ll matrix_pow(ll n){
    --n;
    ll res[2][2]={{1,0},
                   {0,1}};
    ll x[2][2]={{1,2},
                 {1,1}};
    while(n){
        if(n&1)
            mult(res,x);
        mult(x,x);
        n>>=1;
    }
    return (res[0][0]+res[0][1])%INF;
}

int main()
{
    //freopen("/home/lu/文档/r.txt","r",stdin);
    //freopen("/home/lu/文档/w.txt","w",stdout);
    ll n;
    scanf("%lld",&n);
    ll x=matrix_pow(n);
    cout<<((n&1)+x*x)%INF<<endl;
    return 0;
}