首页 > 代码库 > HDU5015-233 Matrix(矩阵快速幂)
HDU5015-233 Matrix(矩阵快速幂)
题目链接
题意:给定一个矩阵的第0列的第1到n个数,第一行第1个数开始每个数分别为233, 2333........,求第n行的第m个数。
思路:将第一行的数全部右移一位,用前一列递推出下一列,构造矩阵,类似如下
1 0 0 0 0 0 0
1 10 0 0 0 0 0
0 1 1 0 0 0 0
0 1 1 1 0 0 0
0 1 1 1 1 0 0
0 1 1 1 1 1 0
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; typedef long long ll; const int N = 15; const int MOD = 10000007; struct mat{ ll s[N][N]; int l; mat(int o) { l = o; memset(s, 0, sizeof(s)); } void init() { s[0][0] = s[1][0] = 1; s[1][1] = 10; for (int i = 2; i < l; i++) for (int j = 1; j <= i; j++) s[i][j] = 1; } mat operator * (const mat& c) { mat ans(l); for (int i = 0; i < l; i++) for (int j = 0; j < l; j++) for (int k = 0; k < l; k++) ans.s[i][j] = (ans.s[i][j] + s[i][k] * c.s[k][j]) % MOD; return ans; } }; int n, m; mat Pow(mat c, int k) { mat ans = c; k--; while (k) { if (k & 1) ans = ans * c; c = c * c; k >>= 1; } return ans; } int main() { while (scanf("%d%d", &n, &m) != EOF) { mat a(n + 2); a.s[0][0] = 3; a.s[1][0] = 233; for (int i = 2; i <= n + 1; i++) scanf("%I64d", &a.s[i][0]); mat tmp(n + 2); tmp.init(); mat ans = Pow(tmp, m); ans = ans * a; printf("%I64d\n", ans.s[n + 1][0]); } return 0; }
HDU5015-233 Matrix(矩阵快速幂)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。