특수 행렬 요소의 합을 찾는 C++ 코드

<시간/>

차원이 n * n인 정방행렬이 주어졌다고 가정합니다. 행렬의 다음 값을 특수 요소라고 합니다 -

주 대각선에 있는 값입니다.
두 번째 대각선에 있는 값입니다.
정확히 (n - 1 / 2) 행이 위에 있고 같은 수의 행이 아래에 있는 행의 값입니다.
왼쪽과 오른쪽에 정확히 (n - 1 / 2)개의 열이 있는 열의 값입니다.

행렬에서 이러한 특수 값의 합을 찾습니다.

따라서 입력이 n =4와 같으면 mat ={{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15 , 16}}, 출력은 107이 됩니다.

단계

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

res := 0
for initialize i := 0, when i < n, update (increase i by 1), do:
   for initialize j := 0, when j < n, update (increase j by 1),
do:
   if i is same as j or i is same as n / 2 or j is same as n/ 2 or i + j is same as n - 1, then:
      res := res + mat[i, j]
print(res)

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다.

#include <bits/stdc++.h>
using namespace std;
#define N 100
void solve(int n, vector<vector<int>> mat) {
   int res = 0;
   for (int i = 0; i < n; i++)
   for (int j = 0; j < n; j++){
      if (i == j || i == n / 2 || j == n / 2 || i + j == n - 1)
         res += mat[i][j];
   }
   cout << res << endl;
}
int main() {
   int n = 4;
   vector<vector<int>> mat = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}};
   solve(n, mat);
   return 0;
}

입력

4, {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}}