Submission #1217912

#	Time	Username	Problem	Language	Result	Execution time	Memory
1217912		dosts	Mosaic (IOI24_mosaic)	C++20	0 / 100	149 ms	30380 KiB

mosaic

#include "mosaic.h"
#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
#define int long long
#define pii pair<int,int>
#define vi vector<int>
#define ff first
#define ss second
#define sp << " " <<
#define all(x) x.begin(),x.end()
#define big(x) ((int)(x.size()))
using namespace std;
const int MOD = 1e9+7, LIM = 1e6+1, inf = 2e9;

vi mosaic(vector<int32_t> X, vector<int32_t> Y,
                              vector<int32_t> T, vector<int32_t> B,
                              vector<int32_t> L, vector<int32_t> R) {
  int N = X.size(),Q = T.size();
  int rows[4][N+1],cols[4][N+1];
  for (int i = 1;i<=N;i++) rows[1][i] = X[i-1];
  for (int i = 1;i<=N;i++) cols[1][i] = Y[i-1];
  for (int i=2;i<=N && i <= 3;i++) rows[i][1] = Y[i-1];
  for (int i=2;i<=N && i<=3;i++) cols[i][1] = X[i-1];
  set<int> diag;
  for (int i = 2;i<=3;i++) {
    for (int j = 2;j<=N;j++) {
      rows[i][j] = (!rows[i-1][j] && !rows[i][j-1]);
      if (rows[i][j]) diag.insert(i-j);
    }
  }
  for (int i = 2;i<=3;i++) {
    for (int j = 2;j<=N;j++) {
      cols[i][j] = (!cols[i-1][j] && !cols[i][j-1]);
      if (cols[i][j]) diag.insert(i-j);
    }
  }
  if (N >= 4) {
    int prv = cols[3][4];
    for (int i = 4;i<=4;i++) {
      for (int j = 4;j<=N;j++) {
        prv = (!rows[3][j] && !prv);
        if (prv) diag.insert(i-j);
      }
    }
    prv = rows[3][4];
    for (int i = 4;i<=4;i++) {
      for (int j = 4;j<=N;j++) {
        prv = (!cols[3][j] && !prv);
        if (prv) diag.insert(i-j);
      }
    }
  }
  vi ans(Q,0);
  for (int i = 0;i<Q;i++) {
    L[i]++,R[i]++,T[i]++,B[i]++;
    if (L[i] != R[i] || T[i] != B[i]) exit(-1);
    if (L[i] <= 3) {
      ans[i] = cols[L[i]][T[i]];
      continue;
    }
    if (T[i] <= 3) {
      ans[i] = rows[T[i]][L[i]];
      continue;
    }
    ans[i] = diag.count(T[i]-L[i]);
  }
  return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 7
• Subtask 4: 0 / 10
• Subtask 5: 0 / 8
• Subtask 6: 0 / 22
• Subtask 7: 0 / 19
• Subtask 8: 0 / 22