2024 ccfcsp认证打卡 2023 05 2 矩阵运算

java 复制代码
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int N = 10010, D = 30;
        long[][] tmp = new long[D][D];  // 用于存储 K 的转置 * V 的结果
        long[][] ans = new long[N][D];  // 存储最终结果
        int n, d;  // n 表示矩阵的行数,d 表示矩阵的列数
        int[][] Q = new int[N][D];  // 输入矩阵 Q
        int[][] K = new int[N][D];  // 输入矩阵 K
        int[][] V = new int[N][D];  // 输入矩阵 V
        int[] W = new int[N];       // 输入数组 W

        n = input.nextInt();  // 输入矩阵行数
        d = input.nextInt();  // 输入矩阵列数
        
        // 输入矩阵 Q
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= d; j++)
                Q[i][j] = input.nextInt();
        
        // 输入矩阵 K
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= d; j++)
                K[i][j] = input.nextInt();
        
        // 输入矩阵 V
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= d; j++)
                V[i][j] = input.nextInt();
        
        // 输入数组 W
        for (int i = 1; i <= n; i++)
            W[i] = input.nextInt();

        // 计算 K 的转置 * V = tmp
        for (int i = 1; i <= d; i++)
            for (int j = 1; j <= d; j++)
                for (int k = 1; k <= n; k++)
                    tmp[i][j] += (long) K[k][i] * V[k][j];

        // 计算 Q * tmp = ans
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= d; j++) {
                for (int k = 1; k <= d; k++)
                    ans[i][j] += (long) Q[i][k] * tmp[k][j];
                ans[i][j] *= (long) W[i];
            }

        // 输出结果
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= d; j++)
                System.out.print(ans[i][j] + " ");
            System.out.println();
        }
    }
}
相关推荐
jiaway2 分钟前
【C语言】第四课 指针与内存管理
c语言·开发语言·算法
菩提树下的凡夫6 分钟前
瑞芯微RV1126目标识别算法Yolov8的部署应用
java·算法·yolo
爱隐身的官人15 分钟前
新后端漏洞(上)- Java RMI Registry反序列化漏洞
java·反序列化漏洞
叫我阿柒啊19 分钟前
从Java全栈到前端框架:一次真实的面试对话与技术解析
java·javascript·typescript·vue·springboot·react·前端开发
晚安里28 分钟前
Spring 框架(IoC、AOP、Spring Boot) 的必会知识点汇总
java·spring boot·spring
爱隐身的官人39 分钟前
新后端漏洞(上)- Aapache Tomcat AJP 文件包含漏洞(CVE-2020-1938)
java·tomcat·ajp
多打代码44 分钟前
2025.09.05 用队列实现栈 & 有效的括号 & 删除字符串中的所有相邻重复项
python·算法
@CLoudbays_Martin111 小时前
为什么动态视频业务内容不可以被CDN静态缓存?
java·运维·服务器·javascript·网络·python·php
四谎真好看1 小时前
Java 学习笔记(进阶篇2)
java·笔记·学习
上官浩仁1 小时前
springboot ioc 控制反转入门与实战
java·spring boot·spring