Cod sursa(job #2295911)

Utilizator TooHappyMarchitan Teodor TooHappy Data 4 decembrie 2018 00:29:51
Problema Elimin Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.78 kb
#include <bits/stdc++.h>
 
using namespace std;
 
ifstream in("elimin.in");
ofstream out("elimin.out");
 
int m, n, r, c, table[1050][1050], precalc[1050], sums[1050], ans = -1e9, minn, maxx, minnTarget, maxxTarget, allSum, sol[20];
 
void solve() {
    int sum = allSum;
    for(int i = 1; i <= maxx; ++i) {
        sums[i] = 0;
    }

    for(int j = 1; j <= minn; ++j) {
        if(!sol[j]) {
            for(int i = 1; i <= maxx; ++i) {
                sums[i] += table[i][j];
            }
        } else {
            sum -= precalc[j];
        }
    }
    sort(sums + 1, sums + maxx + 1);
 
    for(int i = 1; i <= maxxTarget; ++i) {
        sum -= sums[i];
    }
 
    ans = max(ans, sum);
}
 
void bkt(int k, int deleted) {
    if(deleted > minnTarget) {
        return;
    }
    if(k == minn + 1) {
        if(deleted == minnTarget) {
            solve();
        }
    } else {
        for(int i = 0; i <= 1; ++i) {
            sol[k] = i;
            bkt(k + 1, deleted + (i == 1));
        }
    }
}
 
int main() {
    ios::sync_with_stdio(false); in.tie(0); out.tie(0);
 
    in >> m >> n >> r >> c;
    for(int i = 1; i <= m; ++i) {
        for(int j = 1; j <= n; ++j) {
            in >> table[i][j];
        }
    }
 
    if(m < n) {
        minn = m; maxx = n;
        minnTarget = r; maxxTarget = c;
 
        for(int i = 1; i <= m; ++i) {
            for(int j = i + 1; j <= n; ++j) {
                swap(table[i][j], table[j][i]);
            }
        }
    } else {
        minn = n; maxx = m;
        minnTarget = c; maxxTarget = r;
    }

    for(int j = 1; j <= minn; ++j) {
        for(int i = 1; i <= maxx; ++i) {
            precalc[j] += table[i][j];
            allSum += table[i][j];
        }
    }
 
    bkt(1, 0);

    out << ans << "\n";
 
    in.close(); out.close();
 
    return 0;
}