题解 | #矩阵乘法#

矩阵乘法

http://www.nowcoder.com/practice/ebe941260f8c4210aa8c17e99cbc663b

public class Main {
    public static void main(String []args) {
        Scanner in = new Scanner(System.in);
        int m1 = Integer.parseInt(in.nextLine());
        int n = Integer.parseInt(in.nextLine());
        int m2 = Integer.parseInt(in.nextLine());

        int [][] a = new int[m1][n];
        int [][] b = new int[n][m2];
        for (int  i = 0; i < m1; i++) {
            String []strs = in.nextLine().split(" ");
            for (int j = 0; j < n; j++) {
                a[i][j] = Integer.parseInt(strs[j]);
            }
        }
        for (int  i = 0; i < n; i++) {
            String []strs = in.nextLine().split(" ");
            for (int j = 0; j < m2; j++) {
                b[i][j] = Integer.parseInt(strs[j]);
            }
        }
        int[][] c = new int[m1][m2];
        ArrayList<Integer> res = new ArrayList();
        for (int i = 0; i < m1; i++) {
            
            int index = 0;
            while (index < m2) {
                int sum = 0;
                for (int j = 0; j < n; j++) {
                    sum += a[i][j] * b[j][index];
                }
                index++;
                res.add(sum);
            }
        }
        for (int i = 0; i < res.size(); i++) {
            if (i % m2 == 0 && i > 0) {
                System.out.println();
            }
            System.out.print(res.get(i) + " ");
        }
    }
}
全部评论

相关推荐

比亚迪汽车新技术研究院 硬件工程师 总包21左右 硕士
点赞 评论 收藏
分享
昨天 13:08
蚌埠坦克学院 C++
服从性笔试吗,发这么多笔,现在还在发。
蟑螂恶霸zZ:傻 x 公司,发两次笔试,两次部门匹配挂,
投递金山WPS等公司10个岗位 >
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务