Cod sursa(job #1258099)

Utilizator MarcvsHdrMihai Leonte MarcvsHdr Data 8 noiembrie 2014 14:24:36
Problema Plantatie Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.62 kb
#include <iostream>
#include <fstream>
#include <algorithm>

#define MAXN 500

int n, m;
int rqi, rqf, cqi, cqf;
int a[MAXN + 1][MAXN + 1];
int v[1 + 16 * MAXN * MAXN];

int set_elem(int node, int ri, int rf, int ci, int cf) {
  if (ri > rf || ci > cf) return 0;

  if (ri == rf && ci == cf) {
    return v[node] = a[ri][ci];
  }

  int rm = (ri + rf) / 2;
  int cm = (ci + cf) / 2;
  return v[node] = std::max(
      std::max(set_elem(node * 4 - 2, ri, rm, ci, cm),
               set_elem(node * 4 - 1, ri, rm, cm + 1, cf)),
      std::max(set_elem(node * 4 - 0, rm + 1, rf, ci, cm),
               set_elem(node * 4 + 1, rm + 1, rf, cm + 1, cf)));
}

int get_elem(int node, int ri, int rf, int ci, int cf) {
  if (ri > rf || ci > cf) return 0;
  if (rqf < ri || rqi > rf || cqf < ci || cqf > cf) return 0;

  if (rqi <= ri && rqf >= rf && cqi <= ci && cqf >= cf) return v[node];

  int rm = (ri + rf) / 2;
  int cm = (ci + cf) / 2;
  return std::max(
      std::max(
          get_elem(node * 4 - 2, ri, rm, ci, cm),
          get_elem(node * 4 - 1, ri, rm, cm + 1, cf)),
      std::max(
          get_elem(node * 4 - 0, rm + 1, rf, ci, cm),
          get_elem(node * 4 + 1, rm + 1, rf, cm + 1, cf)));
}

int main() {
  std::ifstream in("plantatie.in");
  std::ofstream out("plantatie.out");

  in >> n >> m;
  for (int i = 1; i <= n; ++i) {
    for (int j = 1; j <= n; ++j) {
      in >> a[i][j];
    }
  }
  set_elem(1, 1, n, 1, n);

  for (int i = 0; i < m; ++i) {
    int ri, ci, s;
    in >> rqi >> cqi >> s;
    rqf = rqi + s - 1;
    cqf = cqi + s - 1;
    out << get_elem(1, 1, n, 1, n) << std::endl;
  }

  in.close();
  out.close();

  return 0;
}