原题链接:蓝桥杯算法提高VIP-矩阵乘方
解题思路:
1.在快速幂的基础上修改即可:
1):在矩阵乘法的过程中需要直接求余以降低时间复杂度;
2):矩阵幂的过程中,对于幂次为1和0的输出结果同样要求余,因为这两个分支中并没有运行矩阵乘法的方法,故需要求余。
2.矩阵乘法和矩阵乘方分两个方法写,代码简洁易懂。
注意事项:
python的一些代码特性可以将矩阵乘法在一行内写出!
参考代码:
def matrix_mul(A: list, B: list, p: int) -> list: return [[sum(x_i * y_i % p for x_i, y_i in zip(x_row, y_col)) % p for y_col in zip(*B)] for x_row in A] def matrix_power(matrix: list, power: int, p: int) -> list: if power == 0: res = [[0 for _ in range(2)] for _ in range(2)] for i in range(2): res[i][i] = 1 % p return res elif power == 1: return [[i % p for i in j] for j in matrix] else: res = matrix_power(matrix, power // 2, p) if power & 1: return matrix_mul(matrix, matrix_mul(res, res, p), p) return matrix_mul(res, res, p) b, m = map(int, input().split()) mat = [list(map(int, input().split())) for _ in range(2)] res_ = matrix_power(mat, b, m) for i in range(2): print(*res_[i])
0.0分
1 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复