Cod sursa(job #2310320)

Utilizator GiihuoTihufiNeacsu Stefan GiihuoTihufi Data 31 decembrie 2018 10:54:45
Problema Elimin Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("elimin.in");
ofstream g("elimin.out");

int N,M,R,C;
int A[550][550];

int main()
{
    f>>N>>M>>R>>C;
    if(N<M)
    {
        for(int  i=1;i<=N;i++)
            for(int j=1;j<=M;j++)
                f>>A[i][j],A[0][j]+=A[i][j],A[i][0]+=A[i][j];
    }
    else
    {
        swap(M,N);
        for(int j=1;j<=M;j++)
            for(int i=1;i<=N;i++)
                f>>A[i][j],A[0][i]+=A[i][j],A[j][0]+=A[i][j];
    }

    vector<bool> P(N,0);
    for(int k=0;k<C;k++) P[N-k-1]=1;
    vector<int> SR;
    int smx=0;
    for(int i=1;i<=M;i++) SR.push_back(A[i][0]);
    do
    {
        vector<int> sR(SR.begin(),SR.end());
        for(int j=0;j<N;j++)
            if(P[j])
                for(int i=0;i<M;i++) sR[i]-=A[j+1][i+1];
        sort(sR.begin(),sR.end());
        int sum=0;
        for(int i=R;i<sR.size();i++) sum+=sR[i];
        smx=max(smx,sum);
    }
    while(next_permutation(P.begin(),P.end()));

    g<<smx;

    return 0;
}