POJ - 3233 - Matrix Power Series(矩阵分块 or 分治 + 矩阵快速幂)

2024-02-11 02:18

本文主要是介绍POJ - 3233 - Matrix Power Series(矩阵分块 or 分治 + 矩阵快速幂),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

题目链接:http://poj.org/problem?id=3233

题意:已知一个n*n的矩阵A,和一个正整数k,求S = A + A^2 + A^3 +...+ A^k

分治思路:首先我们知道 A^x 可以用矩阵快速幂求出来。其次可以对 k 进行分治,每次将规模减半,如下:

k = 10S(9) = ( A^1+A^2+A^3+A^4+ A^5 ) + A^5 * ( A^1+A^2+A^3+A^4+A^5 ) = S(5) + A^5 * S(5)

k = 5  : S(5) = ( A^1+A^2 ) + A^3 + A^3 * ( A^1+A^2 ) = S(2) + A^3 + A^3 * S(2)

k = 2   :   S(2) = A^1 + A^2 = S(1) + A^1 * S(1)

从上面几个式子可以发现,当k为奇数或者偶数的区别。

对于一个 k 是偶数则:S(k) = A^{k/2} *S(k/2)+S(k/2)

如果 k 为奇数的话需要加上A^{k/2 + 1}也就是:S(k) = A^{k/2+1}*S(k/2+1)+S(k/2+1)+A^{k/2 + 1}

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 2;
int n, k, MOD;
struct Matrix
{int m[50][50];void init(int x){memset(m, 0, sizeof(m));for(int i = 0; i < n; i++){m[i][i] = x;}}Matrix operator * (const Matrix &a) const{Matrix res;res.init(0);for(int i = 0; i < n; i++)for(int j = 0; j < n; j++)for(int k = 0; k < n; k++)res.m[i][j] = (res.m[i][j] + (m[i][k] * a.m[k][j]) % MOD) % MOD;return res;}Matrix operator + (const Matrix &a) const{Matrix ans;ans.init(0);for(int i = 0; i < n; i++)for(int j = 0; j < n; j++)ans.m[i][j] = (ans.m[i][j] + (m[i][j] + a.m[i][j]) % MOD) % MOD;return ans;}
};Matrix quickPow(Matrix x, int p)
{Matrix res;res.init(1);x = x * res;while(p){if(p & 1) res = res * x;p >>= 1;x = x * x;}return res;
}Matrix a, x;void init()
{for(int i = 0; i < n; i++){for(int j = 0; j < n; j++){scanf("%d", &a.m[i][j]);a.m[i][j] %= MOD;}}
}
Matrix dfs(Matrix c, int k)
{if(k == 1) return c;Matrix dx, dy;dx = dfs(c, k / 2);if(k & 1){dy = quickPow(c, k / 2 + 1);return dx + dy + (dx * dy);}else{dy = quickPow(c, k/2);return dx + (dx * dy);}
}
int main()
{scanf("%d%d%d", &n, &k, &MOD);init();x = dfs(a, k);for(int i = 0; i < n; i++)for(int j = 0; j < n; j++)printf("%d%c",x.m[i][j], j == n - 1 ? '\n' : ' ');return 0;
}
/*
2 2 4
0 1
1 1
------
1 2
2 3
*/

矩阵分块:构造一个2n*2n的矩阵:\begin{bmatrix} A&E\\ zero&E \end{bmatrix}其中E为单位矩阵,zero0矩阵,矩阵的每个元素都是一个矩阵,我们可以把分块矩阵展开,它的乘法和普通矩阵的乘法是一样的。\begin{bmatrix} A&E\\ zero&E \end{bmatrix}^{k+1}=\begin{bmatrix} A ^{k+1}&A^{k}+A^{k-1}+...+A+E\\ zero&E \end{bmatrix}^{k+1}取右上角的矩阵然后减去一个单位矩阵E就是答案。

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 2;
ll n, k, MOD;
struct Matrix
{ll m[80][80];void init(int x){memset(m, 0, sizeof(m));for(int i = 0; i < n * 2; i++){m[i][i] = x;}}Matrix operator * (const Matrix &a) const{Matrix res;res.init(0);for(int i = 0; i < n * 2; i++)for(int j = 0; j < n* 2; j++)for(int k = 0; k < n * 2; k++)res.m[i][j] = (res.m[i][j] + (m[i][k] * a.m[k][j]) % MOD) % MOD;return res;}Matrix operator + (const Matrix &a) const{Matrix ans;ans.init(0);for(int i = 0; i < MAXN; i++)for(int j = 0; j < MAXN; j++)ans.m[i][j] = (ans.m[i][j] + (m[i][j] + a.m[i][j]) % MOD) % MOD;return ans;}
};Matrix quickPow(Matrix x, ll p)
{Matrix res;res.init(1);x = x * res;while(p){if(p & 1) res = res * x;p >>= 1;x = x * x;}return res;
}Matrix a, x;void init()
{for(int i = 0; i < n; i++){for(int j = 0; j < n; j++){scanf("%lld", &a.m[i][j]);a.m[i][j] %= MOD;}a.m[i][i+n] = 1; //构造右上角a.m[i+n][i+n] = 1; //构造左下角}
}
int main()
{while(~scanf("%lld%lld%lld", &n, &k, &MOD)){init();a = quickPow(a, k + 1);for(int i = 0; i < n; i++){for(int j = 0; j < n; j++){x.m[i][j] = a.m[i][j+n];if(i == j)//减去单位矩阵Ex.m[i][j] = (x.m[i][j] - 1 + MOD) % MOD;}}for(int i = 0; i < n; i++)for(int j = 0; j < n; j++)printf("%lld%c",x.m[i][j], j == n - 1 ? '\n' : ' ');}return 0;
}
/*
2 2 4
0 1
1 1
------
1 2
2 3
*/

 

这篇关于POJ - 3233 - Matrix Power Series(矩阵分块 or 分治 + 矩阵快速幂)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/698588

相关文章

电脑桌面文件删除了怎么找回来?别急,快速恢复攻略在此

在日常使用电脑的过程中,我们经常会遇到这样的情况:一不小心,桌面上的某个重要文件被删除了。这时,大多数人可能会感到惊慌失措,不知所措。 其实,不必过于担心,因为有很多方法可以帮助我们找回被删除的桌面文件。下面,就让我们一起来了解一下这些恢复桌面文件的方法吧。 一、使用撤销操作 如果我们刚刚删除了桌面上的文件,并且还没有进行其他操作,那么可以尝试使用撤销操作来恢复文件。在键盘上同时按下“C

poj 3974 and hdu 3068 最长回文串的O(n)解法(Manacher算法)

求一段字符串中的最长回文串。 因为数据量比较大,用原来的O(n^2)会爆。 小白上的O(n^2)解法代码:TLE啦~ #include<stdio.h>#include<string.h>const int Maxn = 1000000;char s[Maxn];int main(){char e[] = {"END"};while(scanf("%s", s) != EO

hdu 2602 and poj 3624(01背包)

01背包的模板题。 hdu2602代码: #include<stdio.h>#include<string.h>const int MaxN = 1001;int max(int a, int b){return a > b ? a : b;}int w[MaxN];int v[MaxN];int dp[MaxN];int main(){int T;int N, V;s

poj 1511 Invitation Cards(spfa最短路)

题意是给你点与点之间的距离,求来回到点1的最短路中的边权和。 因为边很大,不能用原来的dijkstra什么的,所以用spfa来做。并且注意要用long long int 来存储。 稍微改了一下学长的模板。 stack stl 实现代码: #include<stdio.h>#include<stack>using namespace std;const int M

poj 3259 uva 558 Wormholes(bellman最短路负权回路判断)

poj 3259: 题意:John的农场里n块地,m条路连接两块地,w个虫洞,虫洞是一条单向路,不但会把你传送到目的地,而且时间会倒退Ts。 任务是求你会不会在从某块地出发后又回来,看到了离开之前的自己。 判断树中是否存在负权回路就ok了。 bellman代码: #include<stdio.h>const int MaxN = 501;//农场数const int

poj 1258 Agri-Net(最小生成树模板代码)

感觉用这题来当模板更适合。 题意就是给你邻接矩阵求最小生成树啦。~ prim代码:效率很高。172k...0ms。 #include<stdio.h>#include<algorithm>using namespace std;const int MaxN = 101;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int n

poj 1287 Networking(prim or kruscal最小生成树)

题意给你点与点间距离,求最小生成树。 注意点是,两点之间可能有不同的路,输入的时候选择最小的,和之前有道最短路WA的题目类似。 prim代码: #include<stdio.h>const int MaxN = 51;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int P;int prim(){bool vis[MaxN];

poj 2349 Arctic Network uva 10369(prim or kruscal最小生成树)

题目很麻烦,因为不熟悉最小生成树的算法调试了好久。 感觉网上的题目解释都没说得很清楚,不适合新手。自己写一个。 题意:给你点的坐标,然后两点间可以有两种方式来通信:第一种是卫星通信,第二种是无线电通信。 卫星通信:任何两个有卫星频道的点间都可以直接建立连接,与点间的距离无关; 无线电通信:两个点之间的距离不能超过D,无线电收发器的功率越大,D越大,越昂贵。 计算无线电收发器D

poj 1502 MPI Maelstrom(单源最短路dijkstra)

题目真是长得头疼,好多生词,给跪。 没啥好说的,英语大水逼。 借助字典尝试翻译了一下,水逼直译求不喷 Description: BIT他们的超级计算机最近交货了。(定语秀了一堆词汇那就省略吧再见) Valentine McKee的研究顾问Jack Swigert,要她来测试一下这个系统。 Valentine告诉Swigert:“因为阿波罗是一个分布式共享内存的机器,所以它的内存访问

uva 10061 How many zero's and how many digits ?(不同进制阶乘末尾几个0)+poj 1401

题意是求在base进制下的 n!的结果有几位数,末尾有几个0。 想起刚开始的时候做的一道10进制下的n阶乘末尾有几个零,以及之前有做过的一道n阶乘的位数。 当时都是在10进制下的。 10进制下的做法是: 1. n阶位数:直接 lg(n!)就是得数的位数。 2. n阶末尾0的个数:由于2 * 5 将会在得数中以0的形式存在,所以计算2或者计算5,由于因子中出现5必然出现2,所以直接一