java 矩阵的卷积_矩阵快速幂的JAVA实现

本文详细介绍了矩阵快速幂的算法思想,通过将指数表示为二进制形式,利用分治策略有效减少计算次数。文章提供了一个Java实现的例子,展示了如何通过矩阵乘法和位操作来快速计算矩阵的高次幂,大大提高了计算效率。
摘要由CSDN通过智能技术生成

矩阵快速幂

题目描述:

给定一个n*n的矩阵,求该矩阵的k次幂,即P^k。

如何快速的算出一个矩阵的N次幂呢,举个例子,比如A^19 => (A^16)*(A^2)*(A^1)显然采取这样的方式计算时因子数将是log(n)级别的(原来的因子数是n),不仅这样,因子间也是存在某种联系的。 比如A^4能通过(A^2)*(A^2)得到,A^8又能通过(A^4)*(A^4)得到,这点也充分利用了现有的结果作为有利条件。 下面举个例子进行说明:现在要求A^156,而156(10)=10011100(2) 也就有A^156=>(A^4)*(A^8)*(A^16)*(A^128) 考虑到因子间的联系,我们从二进制10011100中的最右端开始计算到最左端。

以下代码就是其实现方式:

public class MatrixMulti {

public static long[][] mut(int k,int n,long[][] A){

long [][] res = new long[n][n];

for(int i = 0 ; i < res.length ;i++){

for(int j = 0 ; j< res[i].length ;j++){

if(i==j){

res[i][j] = 1;

}else{

res[i][j] = 0;

}

}

}

while(k!=0){

if((k&1)==1) res = f(res,A);

k>>=1;

A = f(A,A);

}

return res;

}

public static long[][] f(long[][] A,long[][] B){

long res[][] = new long[A.length][B.length];

for(int i = 0 ; i < res.length ;i++){

for(int j = 0 ; j< res[i].length ;j++){

for(int k = 0 ; k < A[0].length ;k++){

res[i][j] += A[i][k]*B[k][j];

}

}

}

return res;

}

}

其中函数f(),是定义了矩阵的乘法运算,在矩阵中,单位矩阵就是相当于常数1,在函数mut()中,最主要的就是while循环里的代码,运用了分治的思想,快速的计算矩阵的n次幂。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值