Cod sursa(job #2312022)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 3 ianuarie 2019 23:57:43
Problema Elimin Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.89 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
ifstream f("elimin.in");
ofstream g("elimin.out");
int n,m,r,c,sl[8000],sc[8000],s,a[17][8000],v[8000],maxim;
void calcul()
{
    int i,j,d[8000],s1=0,s2=0,val,p;
    for(i=1;i<=m;i++)
    {
        d[i]=sc[i];
    }
    for(i=1;i<=r;i++)
    {
        for(j=1;j<=m;j++)
        {
            d[j]=d[j]-a[v[i]][j];
        }
    }
    for(i=1;i<=r;i++)
    {
        s1=s1+sl[v[i]];
    }
    nth_element(d+1,d+c,d+m+1);
    val=d[c];
    p=c;
    for(i=1;i<=c && p>0;i++)
    {
        if(d[i]<=val)
        {
            s2=s2+d[i];
            p--;
        }
    }
    if(s-s1-s2>maxim)
    {
        maxim=s-s1-s2;
    }
}
void bkt(int k)
{
    int i;
    for(i=v[k-1]+1;i<=n;i++)
    {
        v[k]=i;
        if(k==r)
        {
            calcul();
        }
        else
        {
            bkt(k+1);
        }
    }
}
int main()
{
    int i,j,val,p;
    f>>n>>m>>r>>c;
    if(n<=m)
    {
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                f>>a[i][j];
            }
        }
    }
    else
    {
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                f>>a[j][i];
            }
        }
        swap(n,m);
        swap(r,c);
    }
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)
        {
            sl[i]=sl[i]+a[i][j];
            sc[j]=sc[j]+a[i][j];
            s=s+a[i][j];
        }
    }
    if(r==0)
    {
        nth_element(sc+1,sc+c,sc+m+1);
        val=sc[c];
        p=c;
        for(i=1;i<=m && p>0;i++)
        {
            if(sc[i]<=val)
            {
                s=s-sc[i];
                p--;
            }
        }
        g<<s;
    }
    else
    {
        bkt(1);
        g<<maxim;
    }
    return 0;
}