week14-作业(矩阵快速幂)

数列(矩阵快速幂)

题意

Q老师 对数列有一种非同一般的热爱,尤其是优美的斐波那契数列。

这一天,Q老师 为了增强大家对于斐波那契数列的理解,决定在斐波那契的基础上创建一个新的数列 f(x) 来考一考大家。数列 f(x) 定义如下:

当 x < 10 时,f(x) = x;
当 x ≥ 10 时,f(x) = a0 * f(x-1) + a1 * f(x-2) + a2 * f(x-3) + …… + a9 * f(x-10),ai 只能为 0 或 1。

Q老师 将给定 a0~a9,以及两个正整数 k m,询问 f(k) % m 的数值大小。

聪明的你能通过 Q老师 的考验吗?

输入

输出文件包含多组测试用例,每组测试用例格式如下:

第一行给定两个正整数 k m。(k < 2e9, m < 1e5)

第二行给定十个整数,分别表示 a0~a9。

输出

对于每一组测试用例输出一行,表示 f(k) % m 的数值大小。

输入样例

10 9999
1 1 1 1 1 1 1 1 1 1
20 500
1 0 1 0 1 0 1 0 1 0

输出样例

45
104

思路

构造转移矩阵:
Sn=ASn-1形式
每一项均为10
10维的矩阵,构造后求递推式,容易求得的小解->大解

代码

#include<iostream>
#include<cstring> 
using namespace std;
int k,m;
struct Matrix
{
	int x[12][12];
	Matrix operator*(const Matrix& t)const
	{
		Matrix ret;
		for(int i=0;i<10;i++)
		{
			for(int j=0;j<10;j++)
			{
				ret.x[i][j]=0;
				for(int k=0;k<10;k++)
				{
					ret.x[i][j]+=x[i][k]*t.x[k][j]%m;
					ret.x[i][j]%=m;
			    }
			}
		}return ret;
	}
	Matrix(){
		memset(x,0,sizeof(x));
	}
	Matrix(const Matrix& a)
	{
		memcpy(x,a.x,sizeof(x));
	}
 };
Matrix quick_pow(Matrix a,int x)
{
	Matrix ret;
	for(int i=0;i<10;i++)
	ret.x[i][i]=1;
	while(x)
	{
		if(x&1)
		ret=ret*a;
		a=a*a;
		x>>=1;
	}return ret;

	
}
int main()
{  

    while(cin>>k>>m)
    {  
     if(k<10)
     {
     	cout<<k%m<<endl;continue;
	 }
    Matrix a;
	for(int i=0;i<10;i++)
	{
		int num;cin>>num;a.x[0][i]=num;
	}
	for(int i=1,j=0;i<10;i++,j++)
	a.x[i][j]=1;
	Matrix res=quick_pow(a,k-9);
	long long ans=0;
	for(int i=0;i<10;i++)
	ans=ans+(res.x[0][i]*((9-i)%m));
	ans=ans%m;
	cout<<ans<<endl;
	}return 0;

	
	
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值