求任意方阵的主副对角线之和

思路：若为n阶方阵

主对角线之和	i==j
副对角线	i+j==n-1

#include<bits/stdc++.h>
using namespace std;
int main(){
	int a[100][100];
	int n;
	int sum1=0,sum2=0;//sum1主对角线   sum2副对角线 
	cin>>n;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++)
		cin>>a[i][j];
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			if(i==j)
			sum1+=a[i][j];
			if((i+j)==n-1)
			sum2+=a[i][j]; 	
		}
	}
	cout<<sum1<<endl<<sum2;
	return 0;
	
}