제출 #1221979

#제출 시각아이디문제언어결과실행 시간메모리
1221979hyakup모자이크 (IOI24_mosaic)C++20
7 / 100
82 ms23880 KiB
#include "mosaic.h"
#include <bits/stdc++.h>
using namespace std;

using ll = long long;

const int maxn = 2e5 + 10;

vector<ll> 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();
  vector<ll> sx(n), sy(n), sum(2*maxn), cont(2*maxn), marcx(n), marcy(n);
  sx[0] = sy[0] = x[0];
  for( int i = 1; i < n; i++ ){
    sx[i] = sx[i - 1] + x[i];
    sy[i] = sy[i - 1] + y[i];

    if( i + 1 < n ){
      if( x[i] == 1 && x[i + 1] == 0 ) sum[maxn - i] = -i, cont[maxn - i] = 1, marcx[i] = true;
      else if( i - 2 >= 0 && marcx[i - 2] && x[i + 1] == 0 ) sum[maxn - i] = -i, cont[maxn - i] = 1, marcx[i] = true;

      if( y[i] == 1 && y[i + 1] == 0 ) sum[maxn + i] = i, cont[maxn + i] = 1, marcy[i] = true;
      else if( i - 2 >= 0 && marcy[i - 2] && y[i + 1] == 0 ) sum[maxn + i] = i, cont[maxn + i] = 1, marcy[i] = true;
    }
  }

  for( int i = 1; i < (int)sum.size(); i++ ){
    sum[i] += sum[i - 1];
    cont[i] += cont[i - 1];
  }

  auto query = [&]( ll l, ll r, ll val, ll t ){
    return  val*(cont[r] - cont[l - 1]) + t*(sum[r] - sum[l - 1]);
  };


  vector<ll> resp(q);
  for( int i = 0; i < q; i++ ){
    if( t[i] == 0 && l[i] == 0 ) resp[i] = x[0];
    if( t[i] == 0 ) resp[i] += sx[r[i]] - sx[max(0, l[i] - 1)];
    if( l[i] == 0 ) resp[i] += sy[b[i]] - sy[max(0, t[i] - 1)];

    t[i] = max( t[i], 1 );
    l[i] = max( l[i], 1 );

    ll L = min(b[i] - r[i], t[i] - l[i]);
    ll R = max(b[i] - r[i], t[i] - l[i]);

    resp[i] += query( maxn + L, maxn + R, min(b[i] - t[i] + 1, r[i] - l[i] + 1), 0 );
    resp[i] += query( maxn + R + 1, maxn + b[i] - l[i], b[i] - l[i] + 1, -1 );
    resp[i] += query( maxn + t[i] - r[i], maxn + L - 1, r[i] - t[i] + 1, 1);
  }
  return resp;
}
#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...