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();
        }
    }
}
相关推荐
Scc_hy4 分钟前
强化学习_Paper_1988_Learning to predict by the methods of temporal differences
人工智能·深度学习·算法
巷北夜未央5 分钟前
Python每日一题(14)
开发语言·python·算法
javaisC8 分钟前
c语言数据结构--------拓扑排序和逆拓扑排序(Kahn算法和DFS算法实现)
c语言·算法·深度优先
爱爬山的老虎8 分钟前
【面试经典150题】LeetCode121·买卖股票最佳时机
数据结构·算法·leetcode·面试·职场和发展
SWHL9 分钟前
rapidocr 2.x系列正式发布
算法
XiaoLeisj17 分钟前
【MyBatis】深入解析 MyBatis XML 开发:增删改查操作和方法命名规范、@Param 重命名参数、XML 返回自增主键方法
xml·java·数据库·spring boot·sql·intellij-idea·mybatis
风象南17 分钟前
SpringBoot实现数据库读写分离的3种方案
java·spring boot·后端
振鹏Dong24 分钟前
策略模式——本质是通过Context类来作为中心控制单元,对不同的策略进行调度分配。
java·策略模式
ChinaRainbowSea33 分钟前
3. RabbitMQ 的(Hello World) 和 RabbitMQ 的(Work Queues)工作队列
java·分布式·后端·rabbitmq·ruby·java-rabbitmq
雾月5533 分钟前
LeetCode 914 卡牌分组
java·开发语言·算法·leetcode·职场和发展