Cod sursa(job #3353836)

Utilizator wizardragonWizard Dragon wizardragon Data 12 mai 2026 03:18:27
Problema Elimin Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <bits/stdc++.h>
using namespace std;
int n,m,rr,cc,a[7294],s[7294],sr[15],sol,sumall,o[7294];
bool cmp(int i,int j) {
  return s[i]>s[j];
}
void bkt(int row,int dlt) {
  if(dlt+n-row<rr||sumall<=sol){
    return;
  }
  if(row==n||dlt==rr){
    nth_element(o,o+(m-cc),o+m,cmp);
    int now=0;
    if(m-cc<cc) {
      for(int i=0;i<m-cc;i++){
        now+=s[o[i]];
      }
    }else{
      now=sumall;
      for(int i=m-cc;i<m;i++) {
        now-=s[o[i]];
      }
    }
    sol=max(sol,now);
    return;
  }
  bkt(row+1,dlt);
  if(dlt+1<=rr){
    sumall-=sr[row];
    for(int c=0;c<m;c++) {
      s[c]-=a[row*m+c];
    }
    bkt(row+1,dlt+1);
    sumall+=sr[row];
    for(int c=0;c<m;c++) {
      s[c]+=a[row*m+c];
    }
  }
}
signed main() {
  freopen("elimin.in","r",stdin);
  freopen("elimin.out","w",stdout);
  cin>>n>>m>>rr>>cc;
  if(n<m){
    for(int i=0;i<n;i++){
      for(int j=0;j<m;j++) {
        cin>>a[i*m+j];
      }
    }
  }else{
    for(int i=0;i<n;i++){
      for(int j=0;j<m;j++) {
        cin>>a[j*n+i];
      }
    }
    swap(n,m);
    swap(rr,cc);
  }
  for(int i=0;i<n;i++){
    for(int j=0;j<m;j++){
      sr[i]+=a[i*m+j];
      s[j]+=a[i*m+j];
      sumall+=a[i*m+j];
    }
  }
  for(int i=0;i<m;i++) {
    o[i]=i;
  }
  bkt(0,0);
  cout<<sol<<"\n";
  return 0;
}