给出a+b的值p和ab的值q,问a^n+b^n的值
经过尝试后发现要求出a和b是不现实的,如果都是正数可以二分法求,但a,b可能是负数
那么就在式子上做功夫,a^n+b^n=(a+b)*(a^(n-1)+b^(n-1))-ab*(a^(n-2)+b^(n-2))
设f(n)=a^n+b^n,那么可以得出f(n)=p*f(n-1)-q*f(n-2),另外可以得出 f(0)=2,f(1)=p
接下来用矩阵快速幂求递推式就可以了
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
const int N=2;
struct node
{
ll a[10][10];
};
node shu,ans,mp;
//shu是输入的矩阵,ans是所求答案
node matrix(node x,node y)
{
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++){
mp.a[i][j]=0;
for(int p=1;p<=N;p++)
mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]);
//矩阵乘法
}
return mp;
}
void work(ll k)
{//矩阵快速幂
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
ans.a[i][j]=0;
for(int i=1;i<=N;i++) ans.a[i][i]=1;
node t=shu;
while(k){
if(k&1)
ans=matrix(ans,t);
k>>=1;
t=matrix(t,t);
}
}
int main()
{
ll p,q,n;
while(scanf("%lld%lld%lld",&p,&q,&n)==3) //输入不够三个数才结束
{
if(n==0)
{
printf("2\n");
continue;
}
if(n==1)
{
printf("%lld\n",p);
continue;
}
shu.a[1][1]=p;
shu.a[1][2]=-q;
shu.a[2][1]=1;
shu.a[2][2]=0;
work(n-1);
printf("%lld\n",p*ans.a[1][1]+2*ans.a[1][2]);
}
return 0;
}