输入一个5*5一共25个数字，要求你求出它们两个对角线上的数字之和

题目描述

二维数组真是存矩阵的好东西啊，现在问题来了，输入一个5*5一共25个数字，要求你求出它们两个对角线上的数字之和！

输入

25个数字，5行5列

输出

它们两个对角线上的和

左上->右下第一条

右上->左下第二条

两个数字空格分开

样例输入

1 0 0 0 2
0 1 0 2 0
0 0 1 0 0
0 2 0 1 0
2 0 0 0 1

样例输出

5 9

参考代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll N=1000;

ll a[N][N];

int main()

{

    ll s=0;

    ll n=0;

    for(int i=1; i<=5; i++)

    {

        for(int j=1; j<=5; j++)

        {

            cin>>a[i][j];

        }

    }

    for(int i=1; i<=5; i++)

    {

        for(int j=1; j<=5; j++)

        {

            if(i==j)

            {

                s+=a[i][j];

            }

        }

    }

    for(int m=1; m<=5; m++)

    {

        for(int k=1; k<=5; k++)

        {

            if(m+k==5+1)

            {

                n+=a[m][k];

            }

        }

    }

    cout<<s<<' '<<n;

    return 0;

}