This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#define _CRT_SECURE_NO_WARNINGS
#define fast ios::sync_with_stdio(false); cin.tie(0)
#define foru(i, k, n) for (int i = k; i < n; i++)
#define ford(i, k, n) for (int i = k; i >= n; i--)
#define pb push_back
#define ff first
#define ss second
#include <iostream>
#include <vector>
#include <algorithm>
#include <bitset>
#include <queue>
#include <unordered_set>
#include <set>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int sz = 5e5 + 5;
int n, m, c[sz], nc[sz];
vector<int> a[sz], na[sz], adj[sz];
int lfinal[sz], rfinal[sz], closestL[sz], closestR[sz], lmn[sz], rmx[sz];
short connectedTo[sz];
vector<pii> comps;
vector<int> top;
set<int> st[sz];
void input() {
cin >> n;
foru(i, 0, n - 1) {
cin >> c[i]; c[i]--;
}
foru(i, 0, n) {
int tmp, xx;
cin >> tmp;
while (tmp--) {
cin >> xx;
a[i].pb(--xx);
}
}
}
void computeConnectedTo() {
foru(i, 0, n) {
bool found = 0;
if (i != 0) {
for (int x : a[i]) {
if (x == c[i - 1]) {
connectedTo[i] = -1;
found = 1;
break;
}
}
if (found)continue;
}
if (i != n - 1) {
for (int x : a[i]) {
if (x == c[i]) {
connectedTo[i] = 1;
found = 1;
break;
}
}
if (found)continue;
}
}
}
void computeComps() {
int prvStart = 0;
foru(i, 0, n) {
if (i == n - 1 || connectedTo[i] != 1 || connectedTo[i + 1] != -1) {
comps.pb({ prvStart,i });
prvStart = i + 1;
}
}
m = comps.size();
unordered_set<int> mem;
foru(i, 0, m) {
mem.clear();
foru(j, comps[i].ff, comps[i].ss + 1) {
for (int x : a[j])mem.insert(x);
}
for (int x : mem) {
na[i].pb(x);
st[x].insert(i);
}
if (connectedTo[comps[i].ff] == -1)adj[i - 1].pb(i);
if (connectedTo[comps[i].ss] == 1)adj[i + 1].pb(i);
if (i != m - 1)nc[i] = c[comps[i].ss];
lfinal[i] = rfinal[i] = i;
}
}
void topoSort() {
vector<int> indeg(m, 0);
foru(i, 0, m) {
for (int x : adj[i])indeg[x]++;
}
queue<int> q;
foru(i, 0, m) {
if (indeg[i] == 0)q.push(i);
}
while (!q.empty()) {
int x = q.front(); q.pop();
top.pb(x);
for (int y : adj[x]) {
indeg[y]--;
if (indeg[y] == 0)q.push(y);
}
}
}
void computeClosest() {
for (int i = 0; i < n; i++) {
st[i].insert(-1), st[i].insert(m + 1);
}
for (int i = 0; i < m - 1; i++) {
auto it = st[nc[i]].upper_bound(i);
closestR[i] = (*it);
--it;
closestL[i] = (*it);
}
}
void computeRanges() {
int src;
foru(i, 0, m) {
src = top[i];
if (connectedTo[comps[src].ff] == 0 && connectedTo[comps[src].ss] == 0)continue;
else if (connectedTo[comps[src].ss] == 1) {
if (lfinal[src + 1] <= src) {
lfinal[src] = lfinal[src + 1];
rfinal[src] = rfinal[src + 1];
continue;
}
lfinal[src] = min(lfinal[src], lfinal[src + 1]);
rfinal[src] = max(rfinal[src], rfinal[src + 1]);
}
else if (connectedTo[comps[src].ff] == -1) {
if (rfinal[src - 1] >= src) {
lfinal[src] = lfinal[src - 1];
rfinal[src] = rfinal[src - 1];
continue;
}
lfinal[src] = min(lfinal[src], lfinal[src - 1]);
rfinal[src] = max(rfinal[src], rfinal[src - 1]);
}
while (true) {
if (rfinal[src] != m - 1 && closestL[rfinal[src]] >= lfinal[src]) rfinal[src]++;
else if (lfinal[src] != 0 && closestR[lfinal[src] - 1] <= rfinal[src]) lfinal[src]--;
else break;
}
}
}
void computeFinal() {
foru(i, 0, m) {
foru(j, comps[i].ff, comps[i].ss + 1) {
lmn[j] = comps[lfinal[i]].ff;
rmx[j] = comps[rfinal[i]].ss;
}
}
}
int main() {
fast;
input();
computeConnectedTo();
computeComps();
topoSort();
computeClosest();
computeRanges();
computeFinal();
int q, x, y;
cin >> q;
while (q--) {
cin >> x >> y;
x--; y--;
if (y >= lmn[x] && y <= rmx[x])cout << "YES\n";
else cout << "NO\n";
}
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |