首页 > 代码库 > HDU 5015 233 Matrix 矩阵快速幂
HDU 5015 233 Matrix 矩阵快速幂
链接:http://acm.hdu.edu.cn/showproblem.php?pid=5015
题意:给一个n*m的矩阵(n ≤ 10,m ≤ 109),给出矩阵中的A10,A20,A30...,矩阵中的A01=233,A02=2333,A03=23333...对于Aij来说,Aij=Ai-1j+Aij-1,问矩阵中的Anm是多少。
思路:行数只有10,所以可以进行逐列的递推,进行109的递推需要用矩阵快速幂。
递推矩阵: 初始矩阵:
代码:
#include <algorithm> #include <cmath> #include <cstdio> #include <cstdlib> #include <cstring> #include <ctime> #include <ctype.h> #include <iostream> #include <map> #include <queue> #include <set> #include <stack> #include <string> #include <vector> #define eps 1e-8 #define INF 0x7fffffff #define maxn 10005 #define PI acos(-1.0) #define seed 31//131,1313 typedef long long LL; typedef unsigned long long ULL; using namespace std; #define MOD 10000007 #define maxn 15 #define maxm 15 struct Matrix { int n,m; LL a[maxn][maxm]; void init() { n=m=0; memset(a,0,sizeof(a)); } void change(int c,int d) { n=c; m=d; } Matrix operator +(const Matrix &b) const { Matrix tmp; tmp.n=n; tmp.m=m; for(int i=0; i<n; i++) for(int j=0; j<m; j++) tmp.a[i][j]=a[i][j]+b.a[i][j]; return tmp; } Matrix operator -(const Matrix &b) const { Matrix tmp; tmp.n=n; tmp.m=m; for(int i=0; i<n; i++) for(int j=0; j<m; j++) tmp.a[i][j]=a[i][j]-b.a[i][j]; return tmp; } Matrix operator *(const Matrix &b) const { Matrix tmp; tmp.n=n; tmp.m=b.m; for(int i=0; i<n; i++) { for(int j=0; j<b.m; j++) tmp.a[i][j]=0; } for(int i=0; i<n; i++) for(int j=0; j<b.m; j++) for(int k=0; k<m; k++) { tmp.a[i][j]+=a[i][k]*b.a[k][j]; tmp.a[i][j]%=MOD; } return tmp; } void Copy(const Matrix &x) { n=x.n; m=x.m; for(int i=0; i<n; i++) for(int j=0; j<m; j++) a[i][j]=x.a[i][j]; } }; Matrix M_quick_pow(Matrix m,int k) { Matrix tmp; tmp.n=m.n; tmp.m=m.m;//m=n才能做快速幂 for(int i=0; i<tmp.n; i++) { for(int j=0; j<tmp.n; j++) { if(i==j) tmp.a[i][j]=1; else tmp.a[i][j]=0; } } while(k) { if(k&1) tmp.Copy(tmp*m); k>>=1; m.Copy(m*m); } return tmp; } int main() { #ifdef LOCAL freopen("in.txt", "r", stdin); //freopen("out.txt", "w", stdout); #endif Matrix A,B; int n,m; while(~scanf("%d%d",&n,&m)) { A.init(); B.init(); A.n=n+2; A.m=n+2; A.a[0][0]=10; A.a[0][n+1]=1; for(int i=1; i<=n; i++) { A.a[i][0]=10; for(int j=1; j<=i; j++) A.a[i][j]=1; A.a[i][n+1]=1; } A.a[n+1][n+1]=1; A=M_quick_pow(A,m); B.n=n+2; B.m=1; B.a[0][0]=23; for(int i=1; i<=n; i++) { scanf("%I64d",&B.a[i][0]); B.a[i][0]%=MOD; } B.a[n+1][0]=3; B.Copy(A*B); if(n==0&&m==0) { puts("0"); continue; } printf("%I64d\n",B.a[n][0]); } return 0; }
HDU 5015 233 Matrix 矩阵快速幂
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。