Cod sursa(job #60076)

Utilizator DastasIonescu Vlad Dastas Data 12 mai 2007 12:20:42
Problema Elimin Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.26 kb
#include <cstdio>
//#include <iostream>

#define max 600
#define min 20

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

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

int eliminate[min] = {0};

int s = 0;

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

    if ( m <= n )
    {
//        a = new short*[n];
//        for ( int i = 1; i <= n; ++i )
//            a[i] = new short[m];

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

        int tt = m;
        m = n;
        n = tt;

        tt = r;
        r = c;
        c = tt;
    }
    else
    {
//        a = new short*[m];
//        for ( int i = 1; i <= m; ++i )
//            a[i] = new short[n];

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

    }
}

int partition(int top, int bottom, int arr[])
{
     int x = arr[top];
     int i = top - 1;
     int j = bottom + 1;
     int temp;
     do
     {
        do
        {
            --j;
        } while ( x < arr[j] );

        do
        {
            ++i;
        } while ( x > arr[i] );

        if ( i < j )
        {
            temp = arr[i];
            arr[i] = arr[j];
            arr[j] = temp;
        }
     } while ( i < j );

     return j;
}

void quicksort(int top, int bottom, int arr[])
{
    int middle;
    if ( top < bottom )
    {
        middle = partition(top, bottom, arr);
        quicksort(top, middle, arr);
        quicksort(middle+1, bottom, arr);
    }

    return;
}

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] == 0 )
                b[i] += a[i][j];
        }
    }

    quicksort(1, m, b);

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

    if ( t > s )
        s = t;

    //memset(eliminate, 0, sizeof(eliminate));

    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];

        quicksort(1, m, eliminate);
        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];

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

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

	return 0;
}