C++의 행렬에서 수정된 평균보다 큰 요소 수

<시간/>

행렬의 수정 평균은 다음과 같이 정의됩니다...

(sum(row-wise min) + sum(column-wise max)) / (row_size + column_size)

예를 들어 보겠습니다.

1 2 3
4 5 6
7 8 9

평균 =(합(1 + 4 + 7) + 합(7 + 8 + 9)) / (3 + 3)

먼저 평균을 찾은 다음 평균보다 큰 요소의 수를 계산해야 합니다.

위의 예를 취하면 카운트로 3을 얻습니다. 평균인 6보다 큰 3개의 요소가 있습니다.

알고리즘

매트릭스를 초기화합니다.
행별 최소 요소 합계를 찾습니다.
열별 최대 요소 합계를 찾습니다.
그런 다음 위에서 언급한 공식으로 평균을 구하세요.
이제 행렬의 평균보다 큰 요소의 수를 세십시오.

구현

다음은 위의 알고리즘을 C++로 구현한 것입니다.

#include <bits/stdc++.h>
using namespace std;
#define m 3
#define n 3
int getElementCountGreaterThanMean(int matrix[][n]) {
   int rowSum = 0;
   for (int i = 0; i < m; i++) {
      int min = matrix[i][0];
      for (int j = 1; j < n; j++) {
         if (matrix[i][j] < min){
            min = matrix[i][j];
         }
      }
      rowSum += min;
   }
   int colSum = 0;
   for (int i = 0; i < n; i++) {
      int max = matrix[0][i];
      for (int j = 1; j < m; j++) {
         if (max < matrix[j][i]) {
            max = matrix[j][i];
         }
      }
      colSum += max;
   }
   int mean = (rowSum + colSum) / (m + n);
   int count = 0;
   for (int i = 0; i < m; i++) {
      for (int j = 0; j < n; j++) {
         if (mean < matrix[i][j]) {
            count++;
         }
      }
   }
   return count;
}
int main() {
   int matrix[m][n] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
   cout << getElementCountGreaterThanMean(matrix) << endl;
   return 0;
}

출력

위의 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.