求整数n次方以及矩阵n次方快速解法

项目github地址:bitcarmanlee easy-algorithm-interview-and-practice
欢迎大家star,留言,一起学习进步

1.求整数的n次方

现在想求 m n m^n mn的值,如何用比较快速的方法求得上述值?
例如我们想求 1 2 75 12^{75} 1275的值,快速解法如下:
1.75的二进制数形式为1001011
2. 1 2 75 = 1 2 64 ∗ 1 2 8 ∗ 1 2 2 ∗ 12 12^{75} = 12^{64} * 12^8 * 12 ^ 2 * 12 1275=126412812212
具体求解的时候,我们先计算 1 2 1 12^1 121,然后根据 1 2 1 12^1 121 1 2 2 12^2 122,再根据 1 2 2 12^2 122 1 2 4 12^4 124,以此类推,最后求 1 2 6 4 12^64 1264,即75的二进制数形式总共为多少位,我们就要在原基础上平方几次。这样,就将复杂度为n的计算降到了log(n)。

直接看代码

    public void power() {
	    // 测试4^5的值
        int n = 4, m = 5;
        int power2num = n;
        int result = 1;
        while(m != 0) {
	        // 只有当最低位为1时,结果才乘上现在的值
            if ((m & 1) != 0)
                result *= power2num;
            // 每移位一次,幂方计算一次
            power2num *= power2num;
            m >>= 1;
        }
    }

2.求矩阵的n次方

矩阵n次方的求法与整数n次方的求法思路类似。代码如下

先看两个矩阵相乘的代码

    public long[][] matrixMult(long[][] a, long[][] b) {
        // a的列必须与b的行相等
        assert a.length == b[0].length;
        int n = a.length, m = a[0].length, p = b[0].length;
        long[][] result = new long[n][p];
        //矩阵乘法的基本方式为三层循环
        for(int i=0; i<n; i++) {
            for(int j=0; j<p; j++) {
                for(int k=0; k<m; k++) {
                    result[i][j] += a[i][k] * b[k][j];
                }
            }
        }
        return result;
    }
    public long[][] matrixPower(long[][] matrix, int p) {
        long[][] result = new long[matrix.length][matrix[0].length];

        for(int i=0; i<result.length; i++) {
            result[i][i] = 1;
        }

        long[][] pingfang = matrix;
        for(; p != 0; p >>= 1) {
            if((p & 1) != 0) {
                // 注意result在前面
                result = matrixMult(result, pingfang);
            }
            pingfang = matrixMult(pingfang, pingfang);
        }
        return result;
    }
  • 3
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值