计算矩阵边缘元素之和（java算法优化）

题目描述

输入一个整数矩阵，计算位于矩阵边缘的元素之和。所谓矩阵边缘的元素，就是第一行和最后一行的元素以及第一列和最后一列的元素。

输入格式

第一行分别为矩阵的行数m和列数n（m < 100，n < 100），两者之间以一个空格分开。

接下来输入的m行数据中，每行包含n个整数，整数之间以一个空格分开。

输出格式

输出对应矩阵的边缘元素和。

样例输入

3 3
3 4 1
3 7 1
2 0 1

样例输出

15

代码1

import java.util.Scanner;

public class T2878 {
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int m=sc.nextInt();
        int n=sc.nextInt();
        int[][] arr=new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j]=sc.nextInt();
            }
        }
        int sum=0;
        for (int i = 0; i < n; i++) {
            sum+=(arr[0][i]+arr[m-1][i]);
        }
        for(int i=1;i<n-1;i++){
            sum+=(arr[i][0]+arr[1][n-1]);
        }
        System.out.println(sum);


    }
}

结果实例：

代码运行正确，但是内存过大，运行时间长

优化算法

分析代码可知，当i=0或i=m-1或j=0或j=n-1时符合条件，且互不干扰不重复，因此可写为

i==0||j==0||i==m-1||j==n-1

代码2

import java.util.Scanner;

public class T2878new {

    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int m=sc.nextInt();
        int n=sc.nextInt();
        int[][] arr=new int[m][n];
        int sum=0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j]=sc.nextInt();
                if(i==0||j==0||i==m-1||j==n-1){
                    sum+=arr[i][j];
                }
            }
        }
        System.out.println(sum);
    }
}