首页 > 代码库 > 矩阵经典题目七:Warcraft III 守望者的烦恼(矩阵加速递推)
矩阵经典题目七:Warcraft III 守望者的烦恼(矩阵加速递推)
https://www.vijos.org/p/1067
很容易推出递推式f[n] = f[n-1]+f[n-2]+......+f[n-k]。
构造矩阵的方法:构造一个k*k的矩阵,其中右上角的(k-1)*(k-1)的矩阵是单位矩阵,第k行的每个数分别对应f[n-1],f[n-2],,f[n-k]的系数。然后构造一个k*1的矩阵,它的第i行代表f[i],是经过直接递推得到的。设ans[][]是第一个矩阵的n-k次幂乘上第二个矩阵,f[n]就是ans[k][1]。
注意:用__int64
#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 long long #define _LL __int64 #define eps 1e-12 #define PI acos(-1.0) #define C 240 #define S 20 using namespace std; const int maxn = 15; const int mod = 7777777; int k; struct matrix { _LL mat[maxn][maxn]; void init() { memset(mat,0,sizeof(mat)); for(int i = 1; i <= maxn; i++) mat[i][i] = 1; } }a,b; matrix mul(matrix a, matrix b) { matrix ans; memset(ans.mat,0,sizeof(ans.mat)); for(int i = 1; i <= k; i++) { for(int g = 1; g <= k; g++) { if(a.mat[i][g] == 0) continue; for(int j = 1; j <= k; j++) { ans.mat[i][j] = (ans.mat[i][j] + a.mat[i][g] * b.mat[g][j])%mod; } } } return ans; } matrix pow(matrix a, int n) { matrix ans; ans.init(); while(n) { if(n&1) ans = mul(ans,a); a = mul(a,a); n >>= 1; } return ans; } int main() { int n; while(~scanf("%d %d",&k,&n)) { memset(a.mat,0,sizeof(a.mat)); for(int i = 1; i <= k-1; i++) a.mat[i][i+1] = 1; for(int i = 1; i <= k; i++) a.mat[k][i] = 1; matrix ans = pow(a,n-k); memset(b.mat,0,sizeof(b.mat)); b.mat[0][1] = 1; for(int i = 1; i <= k; i++) { for(int j = 0; j < i; j++) b.mat[i][1] += b.mat[j][1]; } ans = mul(ans,b); printf("%I64d\n",ans.mat[k][1]); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。