求一个3*3矩阵对角线元素之和

#include<iostream>
#include<cstdio>
using namespace std;
int main(){
	int i,j;
	int sum=0;
	int a[3][3];
	for(i=0;i<3;i++){
		for(j=0;j<3;j++){
			cin>>a[i][j];
		}
	}
	for(i=0;i<3;i++){
		for(j=0;j<3;j++){
			if(i+j==2){
				sum+=a[i][j];
			}
		}
	}
	cout<<sum<<endl;
}