Hdu 1757 A Simple Math Problem 矩阵快速幂

最新推荐文章于 2019-08-06 10:39:13 发布

wym_king

最新推荐文章于 2019-08-06 10:39:13 发布

阅读量207

点赞数

CC 4.0 BY-SA版权

分类专栏：数论

本文链接：https://blog.youkuaiyun.com/qq_41603898/article/details/94437601

数论专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种使用矩阵快速幂技术解决特定数学问题的方法，通过定义和操作矩阵结构，实现了对大数运算的高效处理。代码示例展示了如何在C++中实现矩阵乘法和矩阵快速幂算法，用于解决形如Fibonacci数列的递推问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

A Simple Math Problem

//E
#include <bits/stdc++.h>
#define ll long long 
using namespace std;
struct Mat{
	ll m[101][101];
};
int mod;
int n;
Mat a,e;
Mat mul(Mat x,Mat y){
	Mat c;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			c.m[i][j] = 0;
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int k=1;k<=n;k++){
				c.m[i][j] = (c.m[i][j]+ x.m[i][k]*y.m[k][j]%mod)%mod;
			}
		}
	}
	return c;
}
Mat pow(Mat x,ll y){
	Mat ans;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		if(i==j)ans.m[i][i] = 1;
		else ans.m[i][j] = 0;
	while(y){
		if(y&1)ans = mul(ans,x);
		x = mul(x,x);
		y>>=1;
	}
	return ans;
}
int main(){	
	int m; 
	n = 10;
	while(cin>>m>>mod)	
	{       
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				a.m[i][j] = 0;
			}
		}
		for(int i=1;i<=n;i++)            
			cin>>a.m[1][i];              
		for(int i=2;i<=n;i++)           
			a.m[i][i-1]=1;          
			if(m<10) cout<<m%mod;            
			else 		{  
			Mat res;          
			res = pow(a,m-9);            
			int ans=0;            
			for(int i=1;i<=n;i++)                
			ans+=res.m[1][i]*(10-i)%mod;            
			cout<<ans%mod<<endl;        
			}    
	}    
	return 0;
}
/*
10 9999
1 1 1 1 1 1 1 1 1 1
20 500
1 0 1 0 1 0 1 0 1 0

*/