Submission #1238785

#TimeUsernameProblemLanguageResultExecution timeMemory
1238785GabrielMosaic (IOI24_mosaic)C++20
22 / 100
1047 ms2162688 KiB
#include "mosaic.h"
#include <bits/stdc++.h>
using namespace std;
vector< vector<long long> > s;
long long Consulta(int i, int j, int k, int l){
  if(i > k) swap(i, k);
  if(j > l) swap(j, l);
  return s[k][l] - s[i - 1][l] - s[k][j - 1] + s[i - 1][j - 1];
}
vector<long long> mosaic(vector<int> x, vector<int> y, vector<int> t, vector<int> b, vector<int> l, vector<int> r){
  int n = x.size(), q = t.size();
  s.assign(n + 1, vector<long long>(n + 1, 0));
  vector< vector<bool> > Colores(n + 1, vector<bool>(n + 1, 0));
  bool Cero = 1;
  for(int i = 0; i < n; i++){
    Colores[1][i + 1] = x[i] == 1;
    Colores[i + 1][1] = y[i] == 1;
    if(x[i] != 0 or y[i] != 0) Cero = 0;
    s[1][i + 1] = x[i] + s[0][i + 1] + s[1][i] - s[0][i];
    s[i + 1][1] = y[i] + s[i][1] + s[i + 1][0] - s[i][0];
  }
  vector<long long> Respuesta(q);
  if(Cero){
    for(int i = 0; i < q; i++){
      if(t[i] == 0) t[i]++;
      if(l[i] == 0) l[i]++;
      if(t[i] > b[i] or l[i] > r[i]){
        Respuesta[i] = -0;
        continue;
      }
      long long Temporal = (r[i] - l[i] + 1) * ((b[i] - t[i] + 1) / 2);
      if((b[i] - l[i] + 1) % 2 == 1){
        if(t[i] % 2 == l[i] % 2) Temporal += (r[i] - l[i] + 2) / 2;
        else Temporal += (r[i] - l[i] + 1) / 2;
      }
      Respuesta[i] = Temporal;
    }
    return Respuesta;
  }
  for(int i = 2; i <= n; i++){
    for(int j = 2; j <= n; j++){
      Colores[i][j] = Colores[i - 1][j] == 0 and Colores[i][j - 1] == 0;
      s[i][j] = (long long)Colores[i][j] + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
    }
  }
  for(int i = 0; i < q; i++){
    Respuesta[i] = Consulta(t[i] + 1, l[i] + 1, b[i] + 1, r[i] + 1);
  }
  return Respuesta;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...