Cod sursa(job #2415447)

Utilizator usureluflorianUsurelu Florian-Robert usureluflorian Data 25 aprilie 2019 23:48:30
Problema Elimin Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f ("elimin.in");
ofstream g ("elimin.out");
const int nmax=1e3+3;
int n,m,t1,t2,v[nmax][nmax],aux[nmax][nmax],pwp,viz[nmax],sol,s[nmax],t[nmax];
void bulan()
{
    int usu=pwp;
    memset(t,0,sizeof(t));
    for(int i=1;i<=m;++i) usu-=viz[i]*s[i];
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=m;++j) t[i]+=(!viz[j])*v[i][j];
    }
    sort(t+1,t+n+1);
    reverse(t+1,t+n+1);
    for(int i=n;i>n-t1;--i) usu-=t[i];
    sol=max(sol,usu);
}
void solve(int lvl,int act)
{
    if(lvl==t2)
    {
        bulan();
        return;
    }
    for(int i=act;i<=m;++i)
    {
        viz[i]=1;
        solve(lvl+1,i+1);
        viz[i]=0;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    f>>n>>m>>t1>>t2;
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=m;++j) f>>v[i][j];
    }
    if(m>n)
    {
        swap(t1,t2);
        for(int i=1;i<=m;++i)
        {
            for(int j=1;j<=n;++j) aux[i][j]=v[j][m-i+1];
        }
        for(int i=1;i<=m;++i)
        {
            for(int j=1;j<=n;++j) v[i][j]=aux[i][j];
        }
        swap(n,m);
    }
    for(int i=1;i<=m;++i)
    {
        for(int j=1;j<=n;++j)
        {
            pwp+=v[j][i];
            s[i]+=v[j][i];
        }
    }
    solve(0,1);
    g<<sol;
    return 0;
}