Java 算法 矩阵求和

题目描述

经过重重笔试面试的考验，小明成功进入 Macrohard 公司工作。

今天小明的任务是填满这么一张表：

表有 n 行 n 列，行和列的编号都从1算起。

其中第 i 行第 j 个元素的值是 gcd(i, j)的平方，

gcd 表示最大公约数，以下是这个表的前四行的前四列：

1 1 1 1

1 4 1 4

1 1 9 1

1 4 1 16

小明突然冒出一个奇怪的想法，他想知道这张表中所有元素的和。

由于表过于庞大，他希望借助计算机的力量。

输入
一行一个正整数 n 意义见题。

输出
一行一个数，表示所有元素的和。由于答案比较大，请输出模 (10^9 + 7)(即：十亿零七) 后的结果。

输入样例 1

4
输出样例 1

48

解题思路

因为具体的值和其坐标相关，所以可以建立一个二维数组来进行计算，其次就是计算每个位置上最大公因数的平方了，计算好后放入对应的坐标位置，最后把二维数组中的值依次相加之后输出。

代码

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int m=scanner.nextInt();
        m+=1;
        int [][]n=new int[m][m];            //建立矩阵的数组
        int r=0;
        for (int i=1;i<m;i++){
            for (int j=1;j<m;j++){
                n[i][j]=z(i,j);            //计算具体的值后放入数组
            }
        }
        for (int[] o:n){                    //因为是二维数组 所以需要双重循环计算
            for (int p:o){
                r+=p;
            }

        }
        System.out.println(r);              //最后输出矩阵的值
    }
    public static int z(int i,int j){
        int a = Math.max(i, j);             //先求坐标最大公因数
        int []b=new int[a];
        for (int k=1;k<=a;k++){
            if (i%k==0&&j%k==0){
                b[k-1]=k;                   //所有公因数
            }
        }
        Arrays.sort(b);                     //排列
        return (int) Math.pow(b[b.length-1],2);     //最大公因数的平方
    }
}