解题思路:
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.0分

1 人评分

C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:

一点编程也不会写的:零基础C语言学练课程

解决困扰你多年的C语言疑难杂症特性的C语言进阶课程

从零到写出一个爬虫的Python编程课程

只会语法写不出代码?手把手带你写100个编程真题的编程百练课程

信息学奥赛或C++选手的 必学C++课程

蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程

手把手讲解近五年真题的蓝桥杯辅导课程

评论列表 共有 0 条评论

暂无评论