首页 > 代码库 > hdu 5015 233 Matrix(西安网络赛1009)【构造矩阵】
hdu 5015 233 Matrix(西安网络赛1009)【构造矩阵】
说起这题简直醉了。。当时愣是没想到该怎么做,搞了好久,虽然有想过构造矩阵,但是没仔细想下去。
此题构造两个矩阵,假设a[]数组为题目给出的数据,最多有10个元素,我们可以构造一个矩阵A:
a={a[1],a[2],a[3],...a[n],23,3} 大小为1*(n+2)
要得到题目需要的计算结果,那么在构造一个矩阵B,大小为(n+2)*(n+2):(假设n=3)
b= 1 1 1 0 0
0 1 1 0 0
0 0 1 0 0
10 10 10 10 0
1 1 1 1 1
如此,我们计算a*b^m=a*(b^m),这样就能利用矩阵快速幂求解了
#include <iostream> #include <algorithm> #include <cstdio> using namespace std; const long long mod=1e7+7; long long** mul(long long** A,long long** B,long long n,long long m,long long l)//A 为n*m的矩阵,B为m*l的矩阵 { long long **t=new long long*[n]; for(long long i=0;i<n;i++) t[i]=new long long[l]; for(long long i=0;i<n;i++) for(long long j=0;j<l;j++){ long long tmp=0; for(long long k=0;k<m;k++) tmp=(tmp+A[i][k]*B[k][j])%mod; t[i][j]=tmp; } return t; } long long** expo(long long **p,long long k,long long n)//P为n*n的矩阵,k为计算k次幂 { if(k==1) return p; long long **e=new long long*[n]; for(long long i=0;i<n;i++) e[i]=new long long[n]; for(long long i = 0; i < n; ++i) for(long long j = 0; j < n; ++j) e[i][j] = (i == j); while(k) { if(k&1) e=mul(e,p,n,n,n); p=mul(p,p,n,n,n); k>>=1; } return e; } int main() { long long n,m; while(~scanf("%I64d%I64d",&n,&m)) { long long **a=new long long*[1]; a[0]=new long long[n+2]; for(long long i=0;i<n;i++) scanf("%I64d",&a[0][i]); a[0][n]=23,a[0][n+1]=3; long long **b=new long long*[n+2]; for(long long i=0;i<n+2;i++) b[i]=new long long[n+2]; for(long long i=0;i<n+2;i++) for(long long j=0;j<n+2;j++) { if(i<n) { if(i<=j&&j<n) b[i][j]=1; else b[i][j]=0; } else { if(i==n) b[i][j]=10; else b[i][j]=1; } } b[n][n+1]=0; // for(long long i=0;i<n+2;i++) // for(long long j=0;j<n+2;j++) // printf("%d%c",b[i][j],j==n+1?'\n':' '); long long **c; c=expo(b,m,n+2); c=mul(a,c,1,n+2,n+2); printf("%I64d\n",c[0][n-1]); } return 0; }
hdu 5015 233 Matrix(西安网络赛1009)【构造矩阵】
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。