Submission #1244495

#TimeUsernameProblemLanguageResultExecution timeMemory
1244495AlperenT_Mosaic (IOI24_mosaic)C++20
0 / 100
522 ms573176 KiB
#include "mosaic.h"

#include <vector>
#include <bits/stdc++.h> 
#pragma GCC optimize("O3,unroll-loops") 
#pragma GCC target("avx2") 
#define pb push_back
#define F first
#define pii pair<int,int> 
#define all(a) a.begin(),a.end()
#define S second 
#define sz(a) (int)a.size()
#define rep(i , a , b) for(int i = (a) ; i <= (b) ; i++)
#define per(i , a , b) for(int i = (a) ; i >= (b) ; i--)
#define ld double
#define ll long long 
using namespace std ;
const int maxn = 1e6   + 100  , sq = 20 , inf = 1e9 , mod = 998244353;
vector <ll>   a[maxn] , s[maxn] ;
ll fen[maxn]  ,n  ,b[maxn];
int ch(int x, int y){
  if(x+y==0)return 1; 
  return 0 ;
}
void upd(int x, int w){
  x++;
  while(x <= 2*n+3){
    fen[x] += w;
    x += x&-x ;
  }
}
ll que(int x){
  ll ans = 0;
  x++;
  while(x){
    ans += fen[x] ;
    x -= x&-x ;
  }
  return ans; 
}
vector <int> vec[maxn] ; 
ll sm[maxn]; 
std::vector<long long> mosaic(std::vector<int> X, std::vector<int> Y,std::vector<int> T, std::vector<int> B,std::vector<int> L, std::vector<int> R) {
  n = sz(X) ;
  rep(i , 0 ,n-1){
    if(i <= sq+2){
      rep(j , 0, n-1){a[i].pb(0);s[i].pb(0);}
    }else{
      rep(j , 0 ,sq+50){a[i].pb(0);s[i].pb(0);} 
    }
  }
  rep(i , 0 , n-1){
    a[0][i] = X[i] ;
    a[i][0] = Y[i] ;
  }
  rep(i , 1, n-1){
    rep(j ,1, n-1){
      if(i > sq && j > sq)break ;
      if(a[i-1][j]==0 && a[i][j-1]==0){
        a[i][j] = 1; 
      }else{
        a[i][j] = 0 ;
      }
    }
  }
  rep(i , 0, n-1){
    rep(j , 0 ,n-1){
      if(i > sq&& j > sq)break ;
      s[i][j] = a[i][j] ;
      if(i)s[i][j] += s[i-1][j] ;
      if(j)s[i][j] += s[i][j-1];
      if(i&&j)s[i][j] -= s[i-1][j-1] ;
    }
  }
  vector <ll> C ;
  rep(i , 0 , sz(T)-1){
    if(T[i] == 0)T[i]++;
    if(L[i] == 0)L[i]++;
    int s = B[i]-T[i]+1 , y =R[i]-L[i]+1;
    if(s < 0 || y < 0){
      C.pb(0);
      continue ;
    }
    if(s*y%2==0){
      C.pb(s*y/2) ;
    }else{
      if((B[i]+R[i])%2 == 0){
        C.pb(s*y/2+1);
      }else{
        C.pb(s*y/2) ;
      }
    }
  }
  
  return C;
}
#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...