Cod sursa(job #1466376)

Utilizator bciobanuBogdan Ciobanu bciobanu Data 29 iulie 2015 00:56:52
Problema Elimin Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 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 + 1];

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);

  for (int i = r; i < n; i++) {
    v[i] = 1;
  }
  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::make_heap(sum, sum + m);
    q = 0;
    for (int i = 0; i < m - c; i++) {
      q += sum[0];
      std::pop_heap(sum, sum + m - 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;
}