Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++를 사용하여 Matrix에서 최대 합계가 있는 열을 찾습니다.

<시간/>

크기가 M x N인 행렬이 있다고 가정합니다. 최대 합을 갖는 열을 찾아야 합니다. 이 프로그램에서는 까다로운 접근 방식을 따르지 않고 열별로 배열을 탐색한 다음 각 열의 합계를 구하고 합계가 최대값이면 합계와 열 인덱스를 인쇄합니다.

예시

#include<iostream>
#define M 5
#define N 5
using namespace std;
int colSum(int colIndex, int mat[M][N]){
   int sum = 0;
   for(int i = 0; i<M; i++){
      sum += mat[i][colIndex];
   }
   return sum;
}
void maxColumnSum(int mat[M][N]) {
   int index = -1;
   int maxSum = INT_MIN;
   for (int i = 0; i < N; i++) {
      int sum = colSum(i, mat);
      if (sum > maxSum) {
         maxSum = sum;
         index = i;
      }
   }
   cout << "Index: " << index << ", Column Sum: " << maxSum;
}
int main() {
   int mat[M][N] = {
      { 1, 2, 3, 4, 5 },
      { 5, 3, 1, 4, 2 },
      { 5, 6, 7, 8, 9 },
      { 0, 6, 3, 4, 12 },
      { 9, 7, 12, 4, 3 },
   };
   maxColumnSum(mat);
}

출력

Index: 4, Column Sum: 31