Computer >> Máy Tính >  >> Lập trình >> C ++

Mã C ++ để tìm ra tổng các phần tử ma trận đặc biệt

Giả sử, chúng ta được cung cấp một ma trận vuông có kích thước n * n. Các giá trị sau của ma trận được gọi là các phần tử đặc biệt -

  • Các giá trị nằm trong đường chéo chính.

  • Các giá trị nằm trong đường chéo thứ hai.

  • Các giá trị của hàng có chính xác (n - 1/2) hàng phía trên nó và cùng số hàng bên dưới nó.

  • Giá trị của cột có chính xác (n - 1/2) cột ở bên trái và bên phải.

Chúng tôi tìm ra tổng các giá trị đặc biệt này trong ma trận.

Vì vậy, nếu đầu vào là n =4, mat ={{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15 , 16}}, đầu ra sẽ là 107.

Các bước

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

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)

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau đây để hiểu rõ hơn

#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;
}

Đầu vào

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

Đầu ra

107