【CSP：202305-2】矩阵运算（Java）

题目链接

• 202305-2 矩阵运算

题目描述

求解思路

• 调整矩阵运算的顺序：根据矩阵运算符合结合律这一特点，对运算式的顺序进行调整。根据题目给出的数据范围，我们可以先将 $K^T(d\ast n)$ 矩阵和 $V(n\ast d)$ 进行乘法运算。在将 $W$ 矩阵和 $Q(n\ast d)$ 矩阵相乘的结果运算出来，最后将两者做乘法得到最终结果。
• 复杂度分析：
  1. 如果按照顺序直接运算：
     • $W$ 矩阵和 $Q(n\ast d)$ 矩阵相乘：$T_1=O(nd)$。
     • 和 $K^T(d\ast n)$ 矩阵相乘：$T_2=O(n^{2}d)$。
     • 和 $V(n\ast d)$ 矩阵相乘：$T_3=O(n^{2}d)$。
     • 综上，时间复杂度为：$T=O(n^{2}d)$。
  2. 按照交换后的顺序进行运算：
     • $K^T(d\ast n)$ 矩阵和 $V(n\ast d)$ 矩阵相乘$T_1=O(n{d}^2)$。
     • $W$ 矩阵和 $Q(n\ast d)$ 矩阵相乘：$T_2=O(nd)$。
     • 两个运算结果在相乘：$T_3=O(n{d}^2)$。
     • 综上，时间复杂度为：$T=O(n{d}^2)$。
  3. 因为 $n<=10^4$ 而 $d<=20$，因此对于调整运算顺序之后的时间消耗一定比之前要小得多。

实现代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n, d;
        int i, j, k;
        n = in.nextInt();
        d = in.nextInt();
        int[][] Q = new int[n][d];
        int[][] K_T = new int[d][n];
        int[][] V = new int[n][d];
        int[] W = new int[n];

        // 读取Q
        for (i = 0; i < n; i ++) {
            for (j = 0; j < d; j ++) {
                Q[i][j] = in.nextInt();
            }
        }
        // 读取K
        for (i = 0; i < n; i ++) {
            for (j = 0; j < d; j ++) {
                K_T[j][i] = in.nextInt();
            }
        }
        // 读取V
        for (i = 0; i < n; i ++) {
            for (j = 0; j < d; j ++) {
                V[i][j] = in.nextInt();
            }
        }
        // 读取W
        for (i = 0; i < n; i ++) {
            W[i] = in.nextInt();
        }
        // 计算W*Q
        for (i = 0; i < n; i ++) {
            for (j = 0; j < d; j ++) {
                Q[i][j] *= W[i];
            }
        }
        // 计算K_T*V
        long[][] T1 = new long[d][d];
        for (i = 0; i < d; i ++) {
            for (j = 0; j < d; j ++) {
                for (k = 0; k < n; k ++) {
                    T1[i][j] += K_T[i][k] * V[k][j];
                }
            }
        }
        // 计算并输出Q*T1
        long[][] T2 = new long[n][d];
        for (i = 0; i < n; i ++) {
            for (j = 0; j < d; j ++) {
                for (k = 0; k < d; k ++) {
                    T2[i][j] += Q[i][k] * T1[k][j];
                }
                System.out.print(T2[i][j] + " ");
            }
            System.out.println();
        }
    }
}