Submission #1305667

#	Time	Username	Problem	Language	Result	Execution time	Memory
1305667		vako_p	Long Mansion (JOI17_long_mansion)	C++20	10 / 100	3093 ms	19704 KiB

long_mansion

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define pb push_back
#define ff first
#define sd second
#define debug(x) cerr << #x << " -----> " << x << endl;
//#pragma GCC optimize("O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("unroll-loops")

const int mxN = 1e6 + 5;
ll n,l[mxN],r[mxN],lf[mxN],rt[mxN],L[mxN],R[mxN],b[mxN],c[mxN];
vector<ll> a[mxN];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
  cin >> n;
  for(int i = 1; i <= n - 1; i++) cin >> c[i];
  vector<vector<ll>> v(n + 5);
  for(int i = 1; i <= n; i++){
    cin >> b[i];
    a[i].resize(b[i]);
    for(int j = 0; j < b[i]; j++){
      cin >> a[i][j];
      v[a[i][j]].pb(i);
    } 
  }
  for(int i = 1; i <= n - 1; i++){
    bool ok = false;
    r[i] = n + 1;
    auto it = upper_bound(v[c[i]].begin(), v[c[i]].end(), i);
    if(it != v[c[i]].end()) r[i] = *it;
    if(it != v[c[i]].begin()) l[i] = *(--it);
  }
  vector<pair<ll,ll>> vv; 
  for(int i = 1; i <= n - 1; i++){
    lf[i] = n + 1;
    rt[i] = 0;
    if(r[i] < n + 1) for(int j = i + 1; j <= n - 1; j++){
      if(r[i] > j and l[j] <= i){
        rt[i] = j;
      } 
    }
    else{
      rt[i] = n;
    } 
    if(i + 1 <= rt[i]) vv.pb({i + 1, rt[i]});
    if(l[i]) for(int j = i - 1; j >= 1; j--){
      if(l[i] <= j and r[j] > i){
        lf[i] = j;
      } 
    }
    else{
      lf[i] = 0;
    } 
    if(lf[i] + 1 <= i) vv.pb({lf[i] + 1,i});
  }
  vector<ll> nums[2][n + 5];
  for(auto it : vv){
    // cout << it.ff << ' ' << it.sd << endl;
    nums[0][it.sd].pb(it.ff);
    nums[1][it.ff - 1].pb(it.ff);
  } 
  multiset<ll> s;
  s.insert(0);
  for(int i = n; i > 0; i--){
    for(auto it : nums[0][i]) s.insert(it);
    for(auto it : nums[1][i]) s.erase(s.find(it));
    nums[0][i].clear();
    nums[1][i].clear();
    L[i] = *(--s.end());
  }
  s.clear();
  s.insert(n + 1);
  for(auto it : vv){
    nums[0][it.sd].pb(it.sd);
    nums[1][it.ff - 1].pb(it.sd);
  }
  for(int i = n; i > 0; i--){
    for(auto it : nums[0][i]) s.insert(it);
    for(auto it : nums[1][i]) s.erase(s.find(it));
    R[i] = *s.begin();
    // cout << i << ' ' << L[i] << ' ' << R[i] << '\n';
  }
  ll q;
  cin >> q;
  while(q--){
    ll x,y;
    cin >> x >> y;
    if(y >= L[x] and y <= R[x]) cout << "YES" << '\n';
    else cout << "NO" << '\n';
  }
}

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