Cod sursa(job #1466404)

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

#define MAX_N 17
#define MAX_M 7294
#define NIL -1

short a[MAX_M][MAX_N];
bool v[MAX_N];
int sum[MAX_M];

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

  if (n <= m) {
    for (register int i = 0; i < n; ++i) {
      for (register int j = 0; j < m; ++j) {
        fscanf(f, "%hd", &a[j][i]);
      }
    }
  } else {
    for (register int i = 0; i < n; ++i) {
      for (register int j = 0; j < m; ++j) {
        fscanf(f, "%hd", &a[i][j]);
      }
    }
    std::swap(n, m);
    std::swap(r, c);
  }
  fclose(f);
  for (int i = r; i < n; i++) {
    v[i] = 1;
  }
  ans = 0;
  do {
    for (register int i = 0; i < m; ++i) {
      sum[i] = 0;
      for (register int j = 0; j < n; ++j) {
        sum[i] += (a[i][j] & -v[j]);
      }
    }
    std::make_heap(sum, sum + m);
    q = 0;
    for (register int i = 0; i + c < m; ++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;
}