Cod sursa(job #2015428)

Utilizator shantih1Alex S Hill shantih1 Data 26 august 2017 03:27:54
Problema Elimin Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.3 kb
#include <iostream>
#include <fstream>
#include <algorithm>

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

long long i, j, n, m, r, c, k, rez, sum, ss[450], s, sc[450];
int x[450], v[450][450];

long long suma() {
    
    long long mins = 0;
    for (int i = 1; i <= k; i++)
    {
        for (j = 1; j <= m; j++)
        {
            mins += v[x[i]][j];
            ss[j] -= v[x[i]][j];
        }
    }
    
    sort(ss+1, ss+m+1);
    
    for (i = 1; i <= c; i++)
    {   mins += ss[i];  ss[i] = sc[i];  }
    
    for (i = c+1; i <= m; i++)
        ss[i] = sc[i];
    
    return sum - mins;
}

void back(int p) {
    
    long long c;
    for (int i = x[p-1] + 1; i <= n; i++)
    {
        x[p] = i;
        if (p == k)
        {
            c = suma();
            if (c > rez)    rez = c;
        }
        else back(p + 1);
    }
        
}

int main () {
    
    fin >> n >> m >> r >> c;
    for (i = 1; i <= n; i++)
        for (j = 1; j <= m; j++)
        {
            fin >> v[i][j];
            sum += v[i][j];
        }
    
    for (j = 1; j <= m; j++)
        for (i = 1; i <= n; i++)
        {
            sc[j] += v[i][j];
            ss[j] = sc[j];
        }
    
    k = r;
    back(1);
    
    if (r == 0)
        rez = suma();
    
    fout << rez << "\n";
    return 0;
}