Cod sursa(job #60133)

Utilizator DastasIonescu Vlad Dastas Data 12 mai 2007 18:23:25
Problema Elimin Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.35 kb
#include <cstdio>
//#include <iostream>
#include <algorithm>

using namespace std;

#define max 600
#define min 30

FILE *in = fopen("elimin.in","r"), *out = fopen("elimin.out","w");

int n, m, r, c;
short a[max][min];

int eliminate[max] = {0};

int s = 0;

void read()
{
    fscanf(in, "%d %d %d %d", &m, &n, &r, &c);

    if ( m <= n )
    {
        for ( int i = 1; i <= m; ++i )
            for ( int j = 0; j < n; ++j )
                fscanf(in, "%d" , &a[n-j][i]);

        n = n ^ m;
        m = n ^ m;
        n = n ^ m;

        r = r ^ c;
        c = r ^ c;
        r = r ^ c;

    }
    else
    {
        for ( int i = 1; i <= m; ++i )
            for ( int j = 1; j <= n; ++j )
                fscanf(in, "%d" , &a[i][j]);

    }
}

void suma()
{
    int t = 0;
    int b[max] = {0};

    for ( int i = 1; i <= m; ++i )
    {
        for ( int j = 1; j <= n; ++j )
        {
            if ( !eliminate[j] )
                b[i] += a[i][j];
        }
    }

    sort(b+1, b+m+1);

    for ( int i = r+1; i <= m; ++i )
        t += b[i];

    if ( t > s )
        s = t;

    for ( int i = 1; i <= n; ++i )
        eliminate[i] = 0;
}

short st[max] = {0};

void back(int col)
{
    for ( int i = st[col-1]+1; i <= n; ++i )
    {
        st[col] = i;

        if ( col == c )
        {
            for ( int t = 1; t <= c; ++t )
                eliminate[st[t]] = 1;

            suma();
        }
        else
        {
            back(col+1);
        }
    }
}

int main()
{
    read();


    if ( c == 0 && r == 0 )
    {
        for ( int i = 1; i <= m; ++i )
            for ( int j = 1; j <= n; ++j )
                s += a[i][j];
    }
    else if ( c == 0 )
    {
        for ( int i = 1; i <= m; ++i )
            for ( int j = 1; j <= n; ++j )
                eliminate[i] += a[i][j];

        sort(eliminate+1, eliminate+m+1);
        for ( int i = r+1; i <= m; ++i )
            s += eliminate[i];
    }
    else if ( r == 0 )
    {
        for ( int i = 1; i <= m; ++i )
            for ( int j = 1; j <= n; ++j )
                eliminate[j] += a[j][i];

        sort(eliminate+1, eliminate+n+1);
        for ( int i = c+1; i <= n; ++i )
            s += eliminate[i];
    }
    else
        back(1);

    fprintf(out, "%d\n", s);

	return 0;
}