首页 > 代码库 > hdu 5015 233 Matrix(构造矩阵)
hdu 5015 233 Matrix(构造矩阵)
http://acm.hdu.edu.cn/showproblem.php?pid=5015
因为是个二维的递推式,当时没有想到可以这样构造矩阵。从列上看,当前这一列都是由前一列递推得到。根据这一点来构造矩阵。令b[i]代表第i列,是一个(n+2)*1的矩阵,即b[1] = [1,233......],之所以在加了两行,是要从前一个矩阵b[i-1]得到b[i]中的第二个数2333...,再构造一个转换矩阵a,它是一个(n+2)*(n+2)的矩阵,那么a^(m-1) * b就是第m列。
/* a矩阵: 1 0 0 0 0... 3 10 0 0 0... 3 10 1 0 0... 3 10 1 1 0... 3 10 1 1 1... .. b矩阵:第1列 */ #include <stdio.h> #include <iostream> #include <map> #include <set> #include <list> #include <stack> #include <vector> #include <math.h> #include <string.h> #include <queue> #include <string> #include <stdlib.h> #include <algorithm> #define LL __int64 //#define LL long long #define eps 1e-9 #define PI acos(-1.0) using namespace std; const int INF = 0x3f3f3f3f; const int mod = 10000007; struct matrix { LL mat[15][15]; void init() { memset(mat,0,sizeof(mat)); for(int i = 0; i < 15; i++) { mat[i][i] = 1; } } }a,b,res; int n,m; matrix mul(matrix a, matrix b) { matrix ans; memset(ans.mat,0,sizeof(ans.mat)); for(int i = 0; i < n+2; i++) { for(int k = 0; k < n+2; k++) { if(a.mat[i][k] == 0) continue; for(int j = 0; j < n+2; j++) { ans.mat[i][j] += (a.mat[i][k] * b.mat[k][j])%mod; ans.mat[i][j] %= mod; } } } return ans; } matrix pow(matrix a, int n) { matrix ans; ans.init(); while(n) { if(n&1) ans = mul(ans,a); n >>= 1; a = mul(a,a); } return ans; } int main() { int x; while(~scanf("%d %d",&n,&m)) { memset(b.mat,0,sizeof(b.mat)); b.mat[0][0] = 1; b.mat[1][0] = 233; for(int i = 2; i < n+2; i++) { scanf("%d",&x); b.mat[i][0] = (b.mat[i-1][0] + x%mod)%mod; } memset(a.mat,0,sizeof(a.mat)); a.mat[0][0] = 1; for(int i = 1; i < n+2; i++) { a.mat[i][0] = 3; a.mat[i][1] = 10; for(int j = 2; j <= i; j++) a.mat[i][j] = 1; } res = pow(a,m-1); LL anw = 0; for(int i = 0; i < n+2; i++) { anw += (res.mat[n+1][i] * b.mat[i][0])%mod; anw %= mod; } printf("%I64d\n",anw); } return 0; }
hdu 5015 233 Matrix(构造矩阵)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。