Cod sursa(job #1466377)

Utilizator bciobanuBogdan Ciobanu bciobanu Data 29 iulie 2015 01:00:53
Problema Elimin Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <stdio.h>
#include <algorithm>
#include <string.h>

#define MAX_N 7294
#define NIL -1

int a[MAX_N][MAX_N];
bool v[MAX_N];
int sum[MAX_N];
int next[MAX_N];

static inline int buildList(const int &n) {
  int i;
  int prev;

  memset(next, NIL, sizeof(next));

  i = 0;
  while (!v[i]) {
    i++;
  }
  prev = i;
  i++;
  for (; i < n; i++) {
    if (v[i]) {
      next[i] = prev;
      prev = i;
    }
  }
  return prev;
}

int main(void) {
  FILE *f = fopen("elimin.in", "r");
  int n, m, r, c;
  int ans, q;

  fscanf(f, "%d%d%d%d", &n, &m, &r, &c);
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      fscanf(f, "%d", &a[j][i]);
    }
  }
  fclose(f);

  memset(v + r, 1, sizeof(v + r));

  ans = 0;
  do {
    q = buildList(n);
    for (int i = 0; i < m; i++) {
      sum[i] = 0;
      for (int j = q; j != NIL; j = next[j]) {
        sum[i] += a[i][j];
      }
    }
    std::sort(sum, sum + m, std::greater <int>());
    q = 0;
    for (int i = 0; i + c < m; i++) {
      q += sum[i];
    }
    if (q > ans) {
      ans = q;
    }
  } while (std::next_permutation(v, v + n));

  f = fopen("elimin.out", "w");
  fprintf(f, "%d\n", ans);
  fclose(f);
  return 0;
}