URI - 1182 - Coluna na Matriz - Iniciante - C++

Plataforma: URI
Problema: 1182

Enunciado:

Neste problema você deve ler um número que indica uma coluna de uma matriz na qual uma operação deve ser realizada, um caractere maiúsculo, indicando a operação que será realizada, e todos os elementos de uma matriz M[12][12]. Em seguida, calcule e mostre a soma ou a média dos elementos que estão na área verde da matriz, conforme for o caso. A imagem abaixo ilustra o caso da entrada do valor 5 para a coluna da matriz, demonstrando os elementos que deverão ser considerados na operação.

Linguagem: C++

Solução:

#include <iostream>
#include <iomanip>

using namespace std;

int main() {
    int c, i, j;
    char t;
    double m[12][12], a = 0;
    
    cin >> c;
    cin >> t;
    
    for (i = 0; i < 12; i++) {
        for (j = 0; j < 12; j++) {
            cin >> m[i][j];
            if (j == c)
                a = a + m[i][c];
        }
    }
    if (t == 'M')
        a /= 12.0;
    
    cout << fixed << setprecision(1) << a << endl;

    return 0;
}