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

相关文章

使用EasyPoi快速导出Word文档功能的实现步骤

《使用EasyPoi快速导出Word文档功能的实现步骤》EasyPoi是一个基于ApachePOI的开源Java工具库,旨在简化Excel和Word文档的操作,本文将详细介绍如何使用EasyPoi快速... 目录一、准备工作1、引入依赖二、准备好一个word模版文件三、编写导出方法的工具类四、在Export

Python多线程实现大文件快速下载的代码实现

《Python多线程实现大文件快速下载的代码实现》在互联网时代,文件下载是日常操作之一,尤其是大文件,然而,网络条件不稳定或带宽有限时,下载速度会变得很慢,本文将介绍如何使用Python实现多线程下载... 目录引言一、多线程下载原理二、python实现多线程下载代码说明:三、实战案例四、注意事项五、总结引

C#使用Spire.XLS快速生成多表格Excel文件

《C#使用Spire.XLS快速生成多表格Excel文件》在日常开发中,我们经常需要将业务数据导出为结构清晰的Excel文件,本文将手把手教你使用Spire.XLS这个强大的.NET组件,只需几行C#... 目录一、Spire.XLS核心优势清单1.1 性能碾压:从3秒到0.5秒的质变1.2 批量操作的优雅

Mybatis-Plus 3.5.12 分页拦截器消失的问题及快速解决方法

《Mybatis-Plus3.5.12分页拦截器消失的问题及快速解决方法》作为Java开发者,我们都爱用Mybatis-Plus简化CRUD操作,尤其是它的分页功能,几行代码就能搞定复杂的分页查询... 目录一、问题场景:分页拦截器突然 “失踪”二、问题根源:依赖拆分惹的祸三、解决办法:添加扩展依赖四、分页

c++日志库log4cplus快速入门小结

《c++日志库log4cplus快速入门小结》文章浏览阅读1.1w次,点赞9次,收藏44次。本文介绍Log4cplus,一种适用于C++的线程安全日志记录API,提供灵活的日志管理和配置控制。文章涵盖... 目录简介日志等级配置文件使用关于初始化使用示例总结参考资料简介log4j 用于Java,log4c

使用Redis快速实现共享Session登录的详细步骤

《使用Redis快速实现共享Session登录的详细步骤》在Web开发中,Session通常用于存储用户的会话信息,允许用户在多个页面之间保持登录状态,Redis是一个开源的高性能键值数据库,广泛用于... 目录前言实现原理:步骤:使用Redis实现共享Session登录1. 引入Redis依赖2. 配置R

Linux如何快速检查服务器的硬件配置和性能指标

《Linux如何快速检查服务器的硬件配置和性能指标》在运维和开发工作中,我们经常需要快速检查Linux服务器的硬件配置和性能指标,本文将以CentOS为例,介绍如何通过命令行快速获取这些关键信息,... 目录引言一、查询CPU核心数编程(几C?)1. 使用 nproc(最简单)2. 使用 lscpu(详细信

一文详解如何在idea中快速搭建一个Spring Boot项目

《一文详解如何在idea中快速搭建一个SpringBoot项目》IntelliJIDEA作为Java开发者的‌首选IDE‌,深度集成SpringBoot支持,可一键生成项目骨架、智能配置依赖,这篇文... 目录前言1、创建项目名称2、勾选需要的依赖3、在setting中检查maven4、编写数据源5、开启热

C/C++中OpenCV 矩阵运算的实现

《C/C++中OpenCV矩阵运算的实现》本文主要介绍了C/C++中OpenCV矩阵运算的实现,包括基本算术运算(标量与矩阵)、矩阵乘法、转置、逆矩阵、行列式、迹、范数等操作,感兴趣的可以了解一下... 目录矩阵的创建与初始化创建矩阵访问矩阵元素基本的算术运算 ➕➖✖️➗矩阵与标量运算矩阵与矩阵运算 (逐元

MybatisX快速生成增删改查的方法示例

《MybatisX快速生成增删改查的方法示例》MybatisX是基于IDEA的MyBatis/MyBatis-Plus开发插件,本文主要介绍了MybatisX快速生成增删改查的方法示例,文中通过示例代... 目录1 安装2 基本功能2.1 XML跳转2.2 代码生成2.2.1 生成.xml中的sql语句头2