每日作业

矩形运算

设计一个程序: 输入一个3*3的矩形, 求矩形的两个对角线和

样例输入：

1 2 3

4 5 6

7 8 9

样例输出:

15, 15

package com.work;

import java.util.Scanner;

/**
 * @author My
 */
public class Assignment_17 {
    public static void main(String[] args) {
        Scanner reading = new Scanner(System.in);
        System.out.println("输入行数：");
        int m = reading.nextInt();
        System.out.println("输入列数：");
        int n = reading.nextInt();
        int[][] num = new int[m][n];
        System.out.println("输入"+m+"行"+n+"列的数组：");
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                num[i][j] = reading.nextInt();
            }
        }
        int sum1 = 0,sum2 = 0;
        if(m==n){       //行和列相等的情况
            for (int i = 0; i < m; i++) {
                sum1 += num[i][i];
            }
            for (int i = 0; i < n; i++) {
                sum2 += num[i][n-i-1];
            }
        }else {      //行和列不相等的情况
            if (m>n){     //行大于列
                int z=m-n;
                int[][] num2 = new int[m][m];
                for (int i = 0; i < m; i++) {
                    for (int j = 0; j < n; j++) {
                        num2[i][j]=num[i][j];
                    }
                    num2[i][m-z]=0;
                }
                for (int i = 0; i < m; i++) {
                    sum1 += num2[i][i];
                }
                for (int i = 0; i < n; i++) {
                    sum2 += num2[i][n-i-1];
                }
            }else {    //行小于列
                int x=n-m;
                int[][] num2 = new int[n][n];
                for (int i = 0; i < m; i++) {
                    for (int j = 0; j < n; j++) {
                        num2[i][j] = num[i][j];
                    }
                    num2[n-x][i]=0;
                }
                for (int i = 0; i < m; i++) {
                    sum1 += num2[i][i];
                }
                for (int i = 0; i < n; i++) {
                    sum2 += num2[i][n-i-1];
                }
            }
        }
        System.out.println("对角线之和：");
        System.out.println(sum1+" "+sum2);
    }
}