traza de una matriz en c++

//traza de una matriz #include<iostream> using namespace std; main(){ int n; cout<<"ingrese el tamaño del arreglo: ";cin>>n; int A[n][n],B[n]; for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ cout<<"ingrese: ";cin>>A[i][j]; } } for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ if(i==j){ B[i]=A[i][j]; } } } int c=0; for(int i=0;i<n;i++){ c=c+B[i]; } for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ cout<<A[i][j]<<"\t"; } cout<<"\n"; } cout<<"la traza de la matriz es: "<<c; }

programa compilado: