数组/模拟牛客NC.143矩阵乘法

最新推荐文章于 2024-11-14 19:06:32 发布

18阿鲁

最新推荐文章于 2024-11-14 19:06:32 发布

阅读量170

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/weixin_43260719/article/details/111667400

版权

NC.143矩阵乘法

题目描述

给定两个nn的矩阵A和B，求AB。

思路

就是矩阵乘法法则，左行乘右列得一点。
三层循环。

import java.util.*;
public class Solution {
    public int[][] solve (int[][] a, int[][] b) {
        int n = a.length;
        int[][] res = new int[n][n];
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                int temp = 0;
                for(int k = 0; k < n; k++){
                    temp += a[i][k]*b[k][j];
                }
                res[i][j] = temp;
            }
        }
        return res;
    }
}