51nod 分解 1537

题意：

1537 分解
0.5 秒 131,072.0 KB 80 分 5级题
问(1+sqrt(2)) ^n 能否分解成 sqrt(m) +sqrt(m-1)的形式
如果可以 输出 m%1e9+7 否则 输出no

收起
输入
一行，一个数n。（n<=10^18）
输出
一行，如果不存在m输出no，否则输出m%1e9+7
输入样例
2
输出样例
9

思路：

找规律 + 矩阵快速幂

（1）(1+sqrt(2)) ^n 展开之后可以表示为a + b*sqrt(2)，多写几个就发现了！！！
ai = ai-1 + 2 * bi-1;
bi = ai-1 + bi-1;

（2）根据上面的递推式我们就可以开始构造矩阵了

res.m[1][1] = 1; res.m[1][2] = 1;
res.m[2][1] = 2; res.m[2][2] = 1;

只需要乘n-1次就可以得到n次方的结果了

（3）最后输出答案的时候分一下奇数和偶数就好了

我终于学会如何构造矩阵了哈哈哈哈哈哈

代码实现：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
#include<string>
#include<cmath>
#include<cstdlib>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5;
const ll mod = 1e9 + 7;

ll n;
struct node{
	ll m[3][3];
}res;

node mul(node a,node b){
	node tmp1;
	tmp1.m[1][1] = 0; tmp1.m[1][2] = 0;
	tmp1.m[2][1] = 0; tmp1.m[2][2] = 0;
	for(int i = 1;i <= 2;i++){
		for(int j = 1;j <= 2;j++){
			for(int k = 1;k <= 2;k++){
				tmp1.m[i][j] = (tmp1.m[i][j] + ((a.m[i][k] * b.m[k][j]) % mod)) % mod;
			}
		}
	}
	/*for(int i = 1;i <= 2;i++){
		for(int j = 1;j <= 2;j++){
			cout << tmp1.m[i][j] << ' ';
		}
		cout << endl;
	}*/
	return tmp1;
}

node qpow(node a,ll b){
	node tmp;
	tmp.m[1][1] = 1; tmp.m[1][2] = 0;
	tmp.m[2][1] = 0; tmp.m[2][2] = 1;
	while(b){
		if(b & 1) tmp = mul(tmp,a);
		b >>= 1;
		a = mul(a,a);
	}
	return tmp;
}

int main(){
	scanf("%lld",&n);
	res.m[1][1] = 1; res.m[1][2] = 1;
	res.m[2][1] = 2; res.m[2][2] = 1;
	node ans = qpow(res,n - 1);
	/*for(int i = 1;i <= 2;i++){
		for(int j = 1;j <= 2;j++){
			cout << ans.m[i][j] << ' ';
		}
		cout << endl;
	}*/
	node tmp2;
	tmp2.m[1][1] = 1; tmp2.m[1][2] = 1;
	ans = mul(tmp2,ans);
	if(n % 2 == 1){
		ll t = (((ans.m[1][2] * ans.m[1][2]) % mod) * 2) % mod;
		printf("%lld\n",t);
	}
	else{
		ll t = (ans.m[1][1] * ans.m[1][1]) % mod;
		printf("%lld\n",t);
	}
	return 0;
}