编程输入一个n×n矩阵中各元素的值，求出两条对角线上的元素之和

#include<iostream>
using namespace std;
void main()
{    
    int n,i,j,sum1=0,sum2=0,a[100][100];
    cout<<"请输入矩阵的长度n:";cin>>n;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            cin>>a[i][j];
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        {
            if(i==j) sum1+=a[i][j];
            if(i+j==n-1) sum2+=a[i][j];
        }
        cout<<"对角线1元素之和"<<sum1<<endl;
        cout<<"对角线2元素之和"<<sum2<<endl;
}