[HackerRank] Diagonal Difference

출처 : https://www.hackerrank.com/challenges/diagonal-difference/problem

#include <bits/stdc++.h>

using namespace std;

int diagonalDifference(vector<vector<int>> arr) {

    int ans1=0; int ans2=0; int s = arr.size();

    for(int i=0,j=0; i>=0 && i<s;i++,j++) {

        ans1 += arr[i][j]; ans2 += arr[i][s-j-1];

    }

    return abs(ans1-ans2);

}

int main()

{

    ofstream fout(getenv("OUTPUT_PATH"));

    int n;

    cin >> n;

    cin.ignore(numeric_limits<streamsize>::max(), '\n');

    vector<vector<int>> arr(n);

    for (int i = 0; i < n; i++) {

        arr[i].resize(n);

        for (int j = 0; j < n; j++) {

            cin >> arr[i][j];

        }

        cin.ignore(numeric_limits<streamsize>::max(), '\n');

    }

    int result = diagonalDifference(arr);

    fout << result << "\n";

    fout.close();

    return 0;

}