Дано двовимірний масив n*m елементів. Знайти суму елементів головної діагоналі. Нужно...

#include

using namespace std;

#define ll long long

#define ld long double

ll find_sum(vector> a){

ll sum = 0;

for(ll i = 0; i < a.size(); i++)

for(ll j = 0; j < a[i].size(); j++)

if(i == j)

sum += a[i][j];

return sum;

}

signed main(){

ll n,m;

cin >> n >> m;

vector> a(n);

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

vector temp(m);

for(ll j = 0; j < m; j++)

cin >> temp[j];

a[i] = temp;

}

cout << find_sum(a);</strong>

}