Submission #296254

#	Time	Username	Problem	Language	Result	Execution time	Memory
296254		BThero	Long Mansion (JOI17_long_mansion)	C++17	10 / 100	466 ms	14072 KiB

long_mansion

// chrono::system_clock::now().time_since_epoch().count()
#include<bits/stdc++.h>

#define pb push_back
#define eb emplace_back
#define mp make_pair
#define fi first
#define se second
#define all(x) (x).begin(), (x).end()
#define debug(x) cerr << #x << " = " << x << endl;

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;

const int MAXN = (int)5e3 + 5;

vector<pii> bad[MAXN];
vector<int> keys[MAXN];
pii seg[MAXN];
int betw[MAXN];
int A[MAXN], B[MAXN];
int n, q;
int own[MAXN];

void solve() {
  scanf("%d", &n);
  
  for (int i = 1; i < n; ++i) {
    scanf("%d", &betw[i]);
  }
  
  for (int i = 1; i <= n; ++i) {
    int sz;
    scanf("%d", &sz);
    keys[i].resize(sz);
    
    for (int j = 0; j < sz; ++j) {
      scanf("%d", &keys[i][j]);
    }
  }
  
  for (int i = 1; i < n; ++i) {
    int p = i + 1;
    
    while (p <= n && find(all(keys[p]), betw[i]) == keys[p].end()) {
      ++p;
    }
    
    B[i] = p;
  }
  
  for (int i = 2; i <= n; ++i) {
    int p = i - 1;
    
    while (p > 0 && find(all(keys[p]), betw[i - 1]) == keys[p].end()) {
      --p;
    }
    
    A[i] = p;
  }
  
  B[0] = n + 1;
  A[n + 1] = 0;
  
  for (int l = 1; l <= n; ++l) {
    for (int r = l; r <= n; ++r) {
      if (B[l - 1] > r && A[r + 1] < l) {
        bad[r - l + 1].eb(l, r);
      }
    }
  }
  
  for (int i = 1; i <= n; ++i) {
    seg[i] = mp(-1, -1);
  }
  
  for (int len = 1; len <= n; ++len) {
    vector<pii> mark(n + 5, mp(-1, -1));
    
    for (auto it : bad[len]) {
      for (int i = it.fi; i <= it.se; ++i) {
        if (mark[i] != mp(-1, -1)) {
          assert(seg[i] != mp(-1, -1));
        }
      
        mark[i] = it;
      }
    }
    
    for (int i = 1; i <= n; ++i) {
      if (seg[i] == mp(-1, -1)) {
        seg[i] = mark[i];
      }
    }
  }
  
  scanf("%d", &q);
  
  for (int i = 1; i <= q; ++i) {
    int a, b;
    scanf("%d %d", &a, &b);
    
    if (seg[a].fi <= b && b <= seg[a].se) {
      printf("YES\n");
    }
    else {
      printf("NO\n");
    }
  }
}

int main() {
  int tt = 1;
  
  while (tt--) {
    solve();
  }

  return 0;
}

Compilation message (stderr)

long_mansion.cpp: In function 'void solve()':
long_mansion.cpp:28:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   28 |   scanf("%d", &n);
      |   ~~~~~^~~~~~~~~~
long_mansion.cpp:31:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   31 |     scanf("%d", &betw[i]);
      |     ~~~~~^~~~~~~~~~~~~~~~
long_mansion.cpp:36:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   36 |     scanf("%d", &sz);
      |     ~~~~~^~~~~~~~~~~
long_mansion.cpp:40:12: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   40 |       scanf("%d", &keys[i][j]);
      |       ~~~~~^~~~~~~~~~~~~~~~~~~
long_mansion.cpp:99:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   99 |   scanf("%d", &q);
      |   ~~~~~^~~~~~~~~~
long_mansion.cpp:103:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  103 |     scanf("%d %d", &a, &b);
      |     ~~~~~^~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 0 / 15
• Subtask 4: 0 / 75