HDU6198 number number number 矩阵快速幂

题目链接：点击打开链接

题目思路：首先暴力的方法，找出前几个数字，最最原始的暴力，下面附上。前几个数字分别是 4,12,33,88,232,609 我们在比较一下斐波那契数列，发现这几个数字是5,13,34,89,233,610  减去1之后的数字，然后针对后面这个序列，我们可以发现 f(n)=2*(f(n-1)+f(n-2))-f(n-3)  得到相应的f(n)之后，再减去1就是答案啦。

我们就可以构造系数矩阵，第一行 （2,2,-1） 第二行（1,0,0） 第三行（0,1,0）然后利用矩阵快速幂算即可。

一个坑点，我第一次遇到系数矩阵带负数的，所以没考虑 负数取模的情况，所以wa了很多次很多次。负数取模要 先加上mod在取模，这样处理就好啦。

AC代码：

/*
2017年9月10日20:28:54 
HDU6198
矩阵快速幂 签到 
AC 
*/ 
#include <iostream>
#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector> 
using namespace std;
typedef long long ll;
const ll mod=998244353;
const int maxn=3;
/*矩阵快速幂*/
//矩阵存储格式 
 typedef struct{
	ll mat[maxn][maxn];
	void init(){
		memset(mat,0,sizeof(mat));
		for(int i=0;i<maxn;i++){	
			mat[i][i]=1;
		}
	}
}matrix;

//矩阵乘法 
matrix multi(matrix a,matrix b)
{
	matrix c;
	for(int i=0;i<maxn;i++)
	{
		for(int j=0;j<maxn;j++)
		{
			c.mat[i][j]=0;
			for(int k=0;k<maxn;k++)
			{
				//为啥要这么写呢，因为考虑到负数的情况，否则会WA 
				c.mat[i][j]=(c.mat[i][j]+((a.mat[i][k]+mod)%mod *(mod+b.mat[k][j])%mod))%mod;
				//把这两行写成一行速度会快一点  对于HDU6198 从31ms到15ms 
				//c.mat[i][j]%=mod;
			}
		}
	}
	return c;
}
//矩阵快速幂 
matrix fast_mod(ll n,matrix p)
{
	matrix ans,base=p;
	ans.init();
	while(n)
	{
		if(n&1) ans=multi(ans,base);
		base=multi(base,base);
		n>>=1;
	}
	return ans;
}

int main(){
	ll k;
	matrix p;
	memset(p.mat,0,sizeof(p.mat));
	p.mat[0][0]=p.mat[0][1]=2;
	p.mat[0][2]=-1;
	p.mat[1][0]=p.mat[2][1]=1;
	while(~scanf("%lld",&k)){
		if(k==1) printf("4\n");
		else if(k==2) printf("12\n");
		else if(k==3) printf("33\n");
		else {
			ll f1,f2,f3;
			f1=5;
			f2=13;
			f3=34;
			matrix t=fast_mod(k-3,p);
			ll ans=(f3*t.mat[0][0]%mod+f2*t.mat[0][1]%mod+(f1*t.mat[0][2]%mod))%mod;
			printf("%lld\n",ans-1);
		}
	
	}
	
	return 0;
}

暴力求前几项的代码：

#include <iostream>
#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector> 
using namespace std;
typedef long long ll;
const int maxn=100;
ll f[maxn];
bool s[10000];
void init(){
	memset(s,true,sizeof(s));
	f[0]=0;
	f[1]=1;
	for(int i=2;i<=maxn;i++){
		f[i]=f[i-1]+f[i-2];
	}
} 
int main(){
	init();
//	/*  k=6时  6个for循环   筛法的思想去找最小的那个数字 
	for(int i=0;i<=15;i++){
		for(int j=0;j<=15;j++){
			for(int k=0;k<=15;k++){
				for(int m=0;m<=15;m++){
					for(int a=0;a<=15;a++){
						for(int b=0;b<=15;b++){
							s[f[i]+f[j]+f[k]+f[m]+f[a]+f[b]]=false;
						}
						
					}
					
				}
				
			}
			
		}
	}
//	*/
	
//	/*
	for(int i=1;i<=1e4;i++){
		if(s[i]){
			printf("%d",i);
			break;
		}// printf("%d",i);
	}
	//*/
	/*
	for(int i=0;i<maxn;i++){
		printf("%lld\n",f[i]);
		if(i%10==0) printf("\n");
	}
//	*/
	
	return 0;
}