C++实现矩阵的快速幂算法 _ O(logN)级计算矩阵高次方【源码】

胖磊小哥_8443

胖磊小哥_8443

2026-04-10

946人浏览

原创

std::pow对矩阵无效,因其专为标量设计,不支持自定义类型;必须手写矩阵快速幂,以单位矩阵为初始值、自实现矩阵乘法为核心,通过二分迭代将时间复杂度优化至O(log n)。

c++实现矩阵的快速幂算法 _ o(logn)级计算矩阵高次方【源码】

为什么直接用 std::pow 对矩阵无效

因为 std::pow 是为标量(doubleint)设计的,不接受自定义类型或二维数组。C++ 标准库没有内置矩阵类型,更不会自动重载幂运算——你传个 vector<vector>></vector> 进去,编译器直接报错 no matching function for call to 'pow'

必须手写快速幂逻辑,并封装矩阵乘法作为基础操作。

  • 矩阵乘法不能复用 std::multiplies,得自己实现:结果第 i 行第 j 列 = 原矩阵第 i 行与第 j 列的点积
  • 单位矩阵不是全 1,而是对角线为 1、其余为 0 的方阵;它是快速幂的初始“基数”
  • 指数为 0 时必须返回单位矩阵,这点容易漏判,导致 n == 0 时输出全零矩阵

如何写一个通用的矩阵快速幂函数(支持任意大小方阵)

核心是把整数快速幂的思路平移过来:把指数不断右移(n >>= 1),底数平方(mat = mat * mat),遇到奇数位就累积到结果中(res = res * mat)。区别只在“乘法”被替换为矩阵乘法。

下面是一个基于 vector<vector long>></vector> 的简洁实现,支持模运算(防溢出):

C++
C++

"空空如也"

下载
vector<vector long>> mat_mult(const vector<vector long>>& a, const vector<vector long>>& b, long long mod) {
    int n = a.size();
    vector<vector long>> c(n, vector<long long>(n));
    for (int i = 0; i vector<vector long>> mat_pow(vector<vector long>> base, long long exp, long long mod) {
int n = base.size();
vector<vector long>> res(n, vector<long long>(n));
for (int i = 0; i <pre class="brush:php;toolbar:false;">while (exp > 0) {
    if (exp & 1) res = mat_mult(res, base, mod);
    base = mat_mult(base, base, mod);
    exp >>= 1;
}
return res;

}

  • 三重循环顺序是 i-k-j,利于 CPU 缓存局部性;换成 i-j-k 在大矩阵下会明显变慢
  • mod 参数建议始终传入(哪怕为 0LLONG_MAX),避免中间结果溢出——long long 乘两个 1e9 就爆了
  • 如果矩阵固定为 2×2(如斐波那契),可展开循环、用 4 个变量代替 vector,性能提升 3–5 倍

常见错误:维度不匹配、越界、未取模导致 WA

提交 OJ 时最常卡在这几类运行时/答案错误:

  • mat_mult 中没检查 a[0].size() == b.size(),输入非方阵时静默出错(应加断言或抛异常)
  • 初始化 res 时用了 vector(n, vector(n, 1)),结果得到全 1 矩阵,而非单位矩阵
  • 模运算是 (a + b) % mod,但乘法部分写成 a[i][k] * b[k][j] % mod —— 漏了外层括号,先取模再加,导致精度丢失
  • 指数为负数?标准快速幂不处理,需提前判断并返回逆矩阵(一般题目不涉及)

什么时候不该用矩阵快速幂

它只适用于**线性递推关系能表示为固定系数方阵乘法**的场景,比如斐波那契、线性同余生成器、图上路径计数(边权为 1)等。以下情况不适合:

  • 递推式含非线性项(如 f(n) = f(n-1) * f(n-2) + n)——无法写成 A × F(n-1) 形式
  • 系数随 n 变化(如 f(n) = n × f(n-1) + f(n-2))——矩阵 A 不再恒定
  • 矩阵规模 > 100×100 且 exp 不大(如 )——普通快速幂的 <code>O(logN) 优势被 O(N³) 乘法抵消,暴力累乘反而更快

真正省时间的地方,是当 exp 达到 1e18 而矩阵仅 2×2 或 3×3 时——此时乘法开销可忽略,log 次迭代才是关键。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

c++

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.02

5119

3

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.08.29

2625

6

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

2025.08.29

3188

10

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

2025.08.29

2225

10

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

2025.08.29

3188

10

C++中int、float和double的区别
C++中int、float和double的区别

本专题整合了c++中int和double的区别,阅读专题下面的文章了解更多详细内容。

2025.10.23

584

4

function是什么
function是什么

function是函数的意思,是一段具有特定功能的可重复使用的代码块,是程序的基本组成单元之一,可以接受输入参数,执行特定的操作,并返回结果。本专题为大家提供function是什么的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.04

2500

5

js函数function用法
js函数function用法

js函数function用法有:1、声明函数;2、调用函数;3、函数参数;4、函数返回值;5、匿名函数;6、函数作为参数;7、函数作用域;8、递归函数。本专题提供js函数function用法的相关文章内容,大家可以免费阅读。

2023.10.07

434

5

页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

2023.08.14

4656

4

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Conan 2 Essentials 免费课程
Conan 2 Essentials 免费课程

共0课时 | 0人学习

CMake 与 Conan 集成实践
CMake 与 Conan 集成实践

共0课时 | 0人学习

Conan 2 高级依赖模型介绍
Conan 2 高级依赖模型介绍

共0课时 | 0人学习