Cod sursa(job #1842463)

Utilizator Vlad_lsc2008Lungu Vlad Vlad_lsc2008 Data 6 ianuarie 2017 23:39:28
Problema Elimin Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.5 kb
#include<bits/stdc++.h>
using namespace std;

short x[7500][25];
bool v[7500];
int sum[7500];
int r,c,m,n,mini,e1,e2;
long long sol,suma;

void bkt(int poz,int rest)
{
    int i,j;
    if(!rest)
    {
        if(mini==m)
        {
            suma=0;
            for(i=1;i<=n;i++)
            {

                sum[i]=0;
                for(j=1;j<=m;j++)
                    if(!v[j]) sum[i]+=x[j][i];
                suma+=sum[i];

            }
            sort(sum+1,sum+n+1);
            for(i=1;i<=c;i++) suma-=sum[i];
            if(suma>sol) sol=suma;
        }
        else
        {

            suma=0;
            for(i=1;i<=m;i++)
            {
                sum[i]=0;
                for(j=1;j<=n;j++)
                    if(!v[j]) sum[i]+=x[i][j];
                suma+=sum[i];
            }
            sort(sum+1,sum+m+1);
            for(i=1;i<=r;i++) suma-=sum[m+1-i];
            if(suma>sol) sol=suma;

        }

    }
    else
    if(poz<=mini)
    {
        v[poz]=1;
        bkt(poz+1,rest-1);
        v[poz]=0;
        bkt(poz+1,rest);
    }
}

int main()
{
    int i,j;
    freopen("elimin.in","r",stdin);
    freopen("elimin.out","w",stdout);
    scanf("%d%d%d%d",&m,&n,&r,&c);
    for(i=1;i<=m;i++)
        for(j=1;j<=n;j++)
            scanf("%d",&x[i][j]);
    mini=min(m,n);
    if(mini==m) { e1=r; e2=c;}
    else { e1=c; e2=r; }
    bkt(1,e1);
    printf("%d\n",sol);
    fclose(stdin);
    fclose(stdout);
    return 0;
}