Envío 5878

Código

#include<bits/stdc++.h>

using namespace std;



void tests(){
    int n, sum = 0, sum_c = 0; cin>>n;
    bool magic_square = true;
    vector<vector<int>> matrix;
    for (int i = 0; i < n; i++){
        matrix.emplace_back(vector<int> ());
        for (int j = 0; j < n; j++){
            int k; cin>>k;
            matrix[i].push_back(k);
        }
    }

    for(int i = 0; i < n; i ++){
        sum += matrix[i][i];
        sum_c += matrix[i][(n - i) - 1];
    }

    if (sum_c != sum) {
        magic_square = false;
    }

    for (int i = 0; i < n && magic_square; i++){
        int sum_v = 0, sum_h = 0;
        for (int j = 0; j < n; j++){
            sum_v += matrix[i][j];
            sum_h += matrix[j][i];
        }
        if (sum_h != sum || sum_v != sum){
            magic_square = false;
            break;
        }
    }
    
    cout<<(magic_square ? "Yes": "No")<<'\n';


}
int main(){
    tests();
}