矩阵相乘与快速幂运算

矩阵相乘与快速幂运算

矩阵乘法

今天编程计算矩阵乘法，修改了很多次最终结果总是出现一个特别大的数字，十分令我不解

最终才发现原因是 没有声明新的矩阵，在矩阵乘法还没有完成之前，就将原来需要相乘的A、B矩阵给修改了，数字自然会越来越大，最后溢出······

这告诉我使用矩阵乘法必须声明一个新的数组用来存放值。

用Java写会比较简单一些

int[][] mul(int[][] a, int[][] b) { // 矩阵乘法 (m * n)  X (n * k) = m * k 
    int m = a.length, n = a[0].length, k = b[0].length;
    int[][] c = new int[m][k];
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < k; j++) {
            for (int p = 0; p < n; p++) {
                c[i][j] += a[i][p] * b[p][j];
            }
        }
    }
    return c;
}

如果用C++要注意二维数组声明的问题。

假设我们要生成一个4*4的long二维数组。

long** t = new long* [4];
for(int i=0;i<4;i++)
{
    t[i]=new long [4];
}

public int fib(int N) {
    if (N == 0) return 0;
    if (N <= 2) return 1;
    int[][] dp = {{1, 1}}; // dp[2], dp[1]
    int[][] ma = {{1,1},{1,0}};
    N -= 2;
    while (N > 1) {
        if ((N & 1) == 1) dp = mul(dp, ma);
        ma = mul(ma, ma);
        N >>= 1;
    }
    dp = mul(init, ma);
    return dp[0][0]; // dp[n]
}