Envío 2514

Código

#include<iostream>
#include<vector>

using namespace std;

pair<int,int> sumRowCol(vector<vector<int>> &matrix, int p){
    int col =0, row = 0;
    for(int i=0;i<matrix.size();i++){
        row+=matrix[p][i];
        col+=matrix[i][p];
    }
    return {col,row};
}

pair<int,int> sumDiagonals(vector<vector<int>> &matrix){
    int firstDiagonal = 0, secondDiagonal = 0;
    for(int i=0;i<matrix.size();i++){
        firstDiagonal+=matrix[i][i];
        secondDiagonal+=matrix[i][matrix.size()-1-i];
    }
    return {firstDiagonal,secondDiagonal};
}


int main(){
    int n;
    cin>>n;
    vector<vector<int>> matrix;
    matrix.resize(n,vector<int>(n,0));
    for(int i=0;i<n;i++){
        for(int j=0; j<n;j++){
            int val;
            cin>>val;
            matrix[i][j] = val;
        }
    }
    int target = 0;
    for(int i=0;i<matrix[0].size();i++){
        target+=matrix[0][i];
    }
    for(int i=0;i<n;i++){
        pair<int,int> current = sumRowCol(matrix,i);
        if(current.first!=target || current.second!=target){
            cout<<"No\n";
            return 0;
        }
    }
    pair<int,int> diagonals = sumDiagonals(matrix);
    if(diagonals.first!=target || diagonals.second!=target){
        cout<<"No\n";
    }
    else{
        cout<<"Yes\n";
    }
    return 0;
}