Cod sursa(job #2729628)

Utilizator trucker4lifeMoraru Radu-Andrei trucker4life Data 24 martie 2021 23:09:14
Problema Elimin Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.36 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;
ifstream fin("elimin.in");
ofstream fout("elimin.out");

// 1 2 3
// 4 5 6
// 7 8 9

// 1
// 2
// 3

int m, n, r, c;
vector<vector<int>> arr;
vector<int> del;
int ans = -2000000000;

void gen_sol(int used, int start) {
    if (used == r) {
        vector<int> col_sum(n, 0);
        for (int j = 0; j < n; j++)
            for (int i = 0; i < m; i++)
                if (!del[i])
                    col_sum[j] += arr[i][j];

        sort(col_sum.begin(), col_sum.end());
        int sum = 0;
        for (int j = c; j < n; j++)
            sum += col_sum[j];

        ans = max(ans, sum);
    }
    else {
        for (int i = start; i < m; i++) {
            del[i] = 1;
            gen_sol(used+1, start+1);
            del[i] = 0;
        }
    }
}

int main() {
    fin >> m >> n >> r >> c;
    if (m > n) {
        arr = vector<vector<int>>(n, vector<int>(m));
        del = vector<int>(n, 0);

        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                fin >> arr[j][i];
        swap(m, n);
    }
    else {
        arr = vector<vector<int>>(m, vector<int>(n));
        del = vector<int>(m, 0);

        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                fin >> arr[i][j];
    }
    gen_sol(0, 0);
    fout << ans;
}