Cod sursa(job #1753921)

Utilizator Burbon13Burbon13 Burbon13 Data 7 septembrie 2016 12:27:06
Problema Elimin Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.26 kb
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <bitset>
#include <algorithm>

using namespace std;

const int nmx = 7300;

int n,m,r,c; /// r linii , c coloane
int mat[3650][90];
int sum_col[90];
int sol[90];
int sume[3650];
int sumtotal;
int raspuns;
bitset <90> viz;

void citire()
{
    scanf("%d %d %d %d", &n, &m, &r, &c);
    if(n >= m)
    {
        if(m == 1)
        {
            if(c == 1)
            {
                printf("0\n");
            }
            else
            {
                int sum = 0;
                for(int i = 1; i <= n; ++i)
                {
                    int nr;
                    scanf("%d", &nr);
                    sum += nr;
                }
                printf("%d\n", sum);
            }
            exit(0);
        }
        else
        {
            for(int i = 1; i <= n; ++i)
                for(int j = 1; j <= m; ++j)
                    scanf("%d", &mat[i][j]);
        }
    }
    else
    {
        if(n == 1)
        {
            if(r == 1)
            {
                printf("%d\n", 0);
            }
            else
            {
                int sum = 0;
                for(int j = 1; j <= m; ++j)
                {
                    int nr;
                    scanf("%d", &nr);
                    sum += nr;
                }
                printf("%d\n", sum);
            }
            exit(0);
        }
        else
        {
            for(int i = 1; i <= n; ++i)
                for(int j = 1; j <= m; ++j)
                    scanf("%d", &mat[j][i]);
            swap(n,m);
            swap(r,c);
        }
    }
}

void calc_sum_total()
{
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= m; ++j)
            sumtotal += mat[i][j];
}

void calcul_sum_col()
{
    for(int j = 1; j <= m; ++j)
        for(int i = 1; i <= n; ++i)
            sum_col[j] += mat[i][j];
}

void verif()
{
    int sumaux = sumtotal;

    viz.reset();

    for(int i = 1; i <= c; ++i)
        {
            viz[sol[i]] = true;
            sumaux -= sum_col[sol[i]];
        }

    for(int i = 1; i <= n; ++i)
    {
        sume[i] = 0;

        for(int j = 1; j <= m; ++j)
            if(not viz[j])
                sume[i] += mat[i][j];

    }

    int rr = r;

    sort(sume + 1, sume + n + 1);

    int pos = 1;

    while(rr)
    {
        sumaux -= sume[pos];

        -- rr;
        ++ pos;
    }

    raspuns = max(raspuns,sumaux);
}

void combinari()
{
    for(int i = 1; i <= c; ++i)
        sol[i] = 1;

    bool ok = 1;

    while(ok)
    {
        verif();

        bool kaux = 0;

        for(int i = c; i; --i)
            if(sol[i] + 1 + c - i <= m)
            {
                int val = sol[i] + 1;
                for(int j = i; j <= c; ++j)
                    sol[j] = val++;
                kaux = 1;
                break;
            }

        if(not kaux)
            ok = 0;
    }
}

void afish()
{
    printf("%d\n", raspuns);
}

int main()
{
    freopen("elimin.in", "r", stdin);
    freopen("elimin.out", "w", stdout);
    citire();
    calc_sum_total();
    calcul_sum_col();
    combinari();
    afish();
    return 0;
}