-
Notifications
You must be signed in to change notification settings - Fork 18
/
Copy path2143.cpp
130 lines (113 loc) · 3.89 KB
/
2143.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
using namespace std;
#define print_op(...) ostream& operator<<(ostream& out, const __VA_ARGS__& u)
template<typename A, typename B> print_op(pair<A, B>) { return out << "(" << u.first << ", " << u.second << ")"; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> print_op(T_container) { out << "{"; string sep; for (const T &x : u) out << sep << x, sep = ", "; return out << "}"; }
template<typename T> void dbg_out(string s, T x) {cerr << "\033[1;35m" << s << "\033[0;32m = \033[33m" << x << "\033[0m\n";}
template<typename T, typename... Args> void dbg_out(string s, T x, Args... args) {for (int i=0, b=0; i<(int)s.size(); i++) if (s[i] == '(' || s[i] == '{') b++; else
if (s[i] == ')' || s[i] == '}') b--; else if (s[i] == ',' && b == 0) {cerr << "\033[1;35m" << s.substr(0, i) << "\033[0;32m = \033[33m" << x << "\033[31m | "; dbg_out(s.substr(s.find_first_not_of(' ', i + 1)), args...); break;}}
#ifdef LOCAL
#define dbg(...) dbg_out(#__VA_ARGS__, __VA_ARGS__)
#else
#define dbg(...)
#endif
using ll = long long;
using ld = long double;
#define ar array
#define sz(x) ((int)x.size())
#define rep(i, a, b) for (int i = (int)(a); i < (int)(b); i++)
#define all(a) (a).begin(), (a).end()
const int MAX_N = 5e4;
const int MAX_L = 20;
const int MAX_C = 26;
const ll MOD = 1e9 + 7;
const ll INF = 1e9;
const ld EPS = 1e-9;
struct SCC {
int n, ti;
vector<int> num, id, stk;
vector<vector<int>> adj, dag, comp;
vector<bitset<MAX_N>> dp;
SCC(int _n) : n(_n), ti(0), num(n), id(n, -1), adj(n) {}
void addEdge(int u, int v) {
adj[u].push_back(v);
}
void init() {
for (int u = 0; u < n; u++)
if (!num[u])
dfs(u);
dag.resize(comp.size());
for (auto &c : comp)
for (int u : c)
for (int v : adj[u])
if (id[u] != id[v])
dag[id[u]].push_back(id[v]);
for (auto &v : dag) {
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
}
int m = sz(comp);
dp.resize(m);
vector<bool> vis(m);
for (auto &c : comp)
for (int u : c)
dp[id[u]].set(u);
function<void(int)> dfsDP = [&](int u) {
vis[u] = true;
for (int v : dag[u]) {
if (!vis[v]) dfsDP(v);
dp[u] |= dp[v];
}
};
for (int i = 0; i < m; i++)
if (!vis[i])
dfsDP(i);
}
bool query(int u, int v) {
return dp[id[u]][v];
}
int dfs(int u) {
int low = num[u] = ++ti;
stk.push_back(u);
for (int v : adj[u]) {
if (!num[v])
low = min(low, dfs(v));
else if (id[v] == -1)
low = min(low, num[v]);
}
if (low == num[u]) {
comp.emplace_back();
do {
id[stk.back()] = (int) comp.size() - 1;
comp.back().push_back(stk.back());
stk.pop_back();
} while (comp.back().back() != u);
}
return low;
}
};
// Compress the graph into a DAG through SCC, then bitset optimisation DP
void solve(int tc = 0) {
int n, m, q; cin >> n >> m >> q;
SCC scc(n);
for (int i = 0; i < m; i++) {
int u, v; cin >> u >> v; u--; v--;
scc.addEdge(u, v);
}
scc.init();
while (q--) {
int u, v; cin >> u >> v; u--; v--;
cout << (scc.query(u, v) ? "YES" : "NO") << "\n";
}
}
signed main() {
ios_base::sync_with_stdio(false); cin.tie(NULL);
int tc = 1;
// cin >> tc;
for (int t = 1; t <= tc; t++) {
// cout << "Case #" << t << ": ";
solve(t);
}
}