阿里笔试编程题 Floyd+快速幂

最新推荐文章于 2022-02-09 20:14:24 发布

cylzty110

最新推荐文章于 2022-02-09 20:14:24 发布

阅读量310

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/cylzty110/article/details/81067008

本文详细介绍了一种矩阵乘法的实现方式，并通过算法优化提高了运算效率。代码使用Java语言编写，展示了如何处理负数元素并进行矩阵相加运算。此外，文章还介绍了如何利用循环和条件判断来优化矩阵乘法过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.Scanner;

public class Main {

    public static int[][] multiply(int[][] matrix1,int[][] matrix2,int size){
        int[][] result = new int[size][size];
        for(int i = 0; i < size; i++){
            for(int j = 0; j < size; j++){
                result[i][j] = -1;
            }
        }

        for(int k = 0; k < size; k++){
            for(int i = 0; i < size; i++){
                for(int j = 0; j < size; j++){
                    if(matrix1[i][k] < 0 || matrix2[k][j] < 0) continue;
                    if(result[i][j] < 0 || result[i][j] > matrix1[i][k] + matrix2[k][j]){
                        result[i][j] = matrix1[i][k] + matrix2[k][j];
                    }
                }
            }
        }
        return result;
    }


    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n =scan.nextInt();
        int m = scan.nextInt();
        int[][] number=new int[n][n];
        int[][] result;
        int[][] ans;
        for(int i=0;i<n;i++){
            for (int j = 0; j < n; j++){
                number[i][j]=scan.nextInt();
                if(number[i][j] == 0) number[i][j] = -1;
            }
        }
        result = ans = number;
        m--;
        while(m > 0){
            if(m%2==1){
                result = multiply(result,ans,n);
            }
            ans = multiply(ans,ans,n);
            m = m/2;
        }

        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                System.out.print(result[i][j] + ",");
            }
            System.out.println();
        }


    }

}