Cod sursa(job #2783066)

Utilizator bogdanvladmihaiBogdan Vlad-Mihai bogdanvladmihai Data 13 octombrie 2021 18:48:27
Problema Elimin Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <bits/stdc++.h>

#define pb push_back

using namespace std;

int n, m, r, c;

int maxSum;

int a[20][7300];

int s[maxN + 5];

int st[maxN + 5];

void bkt(int p, int sumAll) {
  for (int i = st[p - 1] + 1; i <= n; i++) {
    st[p] = i;
    for (int j = 1; j <= m; j++) {
      s[j] -= a[i][j];
      sumAll -= a[i][j];
    }
    if (p == r) {
      vector<int> cs;
      for (int j = 1; j <= m; j++) {
        cs.pb(s[j]);
      }
      sort(cs.begin(), cs.end());
      int sum = 0;
      for (int j = c; j < m; j++) {
        sum += cs[j];
      }
      maxSum = max(maxSum, sum);
    } else {
      if (sumAll >= maxSum) {
        bkt(p + 1, sumAll);
      }
    }
    for (int j = 1; j <= m; j++) {
      s[j] += a[i][j];
      sumAll += a[i][j];
    }
  }
}

int main() {
  freopen("elimin.in", "r", stdin);
  freopen("elimin.out", "w", stdout);
  scanf("%d %d %d %d", &n, &m, &r, &c);
  if (n >= m) {
    for (int i = 1; i <= n; i++) {
      for (int j = 1; j <= m; j++) {
        scanf("%d", &a[j][i]);
      }
    }
    swap(n, m);
    swap(r, c);
  } else {
    for (int i = 1; i <= n; i++) {
      for (int j = 1; j <= m; j++) {
        scanf("%d", &a[i][j]);
      }
    }
  }
  int sumAll = 0;
  for (int i = 1; i <= n; i++) {
    for (int j = 1; j <= m; j++) {
      s[j] += a[i][j];
      sumAll += a[i][j];
    }
  }
  bkt(1, sumAll);
  printf("%d\n", maxSum);
  return 0;
}