CF450B - Jzzhu and Sequences

看了答案才知道有找规律的做法.但为了锻炼自己还是写了矩阵快速幂
这是一个模板题,挺好的.帮助复习了矩阵乘法.
有几个点,第一个是n=1的时候要特判
因为n=3的时候相当于第一次矩阵乘法.所以要快速幂时要quick(n-2)
怎么找用来累乘的二阶矩阵应该挺简单的吧?不会的也可以评论区留言.写了这么多博客还没有评论的卑微作者在线解答.

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;


ll a[2][2]={1,-1,1,0},b[2][2]={1,0,0,1},c[2][2];
const ll M =1e9+7; 

void J(ll a1[2][2],ll b1[2][2]){
	memset(c,0,sizeof(c));
	for(int k = 0;k < 2;k++)
	  for(int i = 0;i < 2;i++)
	     for(int j =0 ;j < 2;j++)
	      {
		  c[i][j] += (a1[i][k]*b1[k][j])%M;
          c[i][j]%=M;
		  }
	memcpy(a1,c,sizeof(c));
}

void quick(ll n1){
   
	while(n1)
	{
		if (n1&1)
	 	 {
	 	 	J(b,a);   	  
		 }
		 J(a,a);	
		 n1>>=1;
	 	 	      	
	}
}
ll p,q,n;
int main(){
	cin>>p>>q;
	cin>>n;
    if(n==1) cout<<(p+M)%M<<endl;
	else {
	quick(n-2);
    
	ll sum = (b[0][0]*q+b[0][1]*p)%M;
	if(sum<0) cout<<sum+M<<endl;
	else cout<<sum<<endl;
}
 return 0; 
}