题目 1024: [编程入门]矩阵对角线求和

题目描述:

求一个3×3矩阵对角线元素之和。

代码:

package lanqiao;

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        int [][]arr = new int[3][3];
        int sum1 = 0;
        int sum2 = 0;

        //arr.length求矩阵的行数
        //arr[i].length求每行包含几列
        Scanner sc = new Scanner(System.in);

        for(int i = 0;i < arr.length;i ++)
        {
            for(int j = 0;j < arr[i].length;j ++)
            {
                arr[i][j] = sc.nextInt();

                if(i == j)
                {
                    sum1 += arr[i][j];
                }
                if(i + j == 2)
                {
                    sum2 += arr[i][j];
                }
            }
        }
        System.out.println(sum1 + " " + sum2);
    }
}