Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- int main()
- {
- ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
- int n, m;
- cin >> n >> m;
- vector<pii> e(m);
- vector<vi> g(n), gt(n);
- vi dp(n), rdp(n);
- for (int i = 0; i < m; ++i)
- {
- int x, y;
- cin >> x >> y;
- --x, --y;
- g[x].inb(y);
- gt[y].inb(x);
- e[i] = mk(x, y);
- }
- vi was(n);
- function<void(int)> dfs = [&](int u)
- {
- was[u] = 1;
- dp[u] = 1;
- for (int to : g[u])
- {
- if (!was[to])
- dfs(to);
- dp[u] += dp[to];
- }
- };
- for (int i = 0; i < n; ++i)
- {
- if (!was[i])
- dfs(i);
- }
- was.assign(n, 0);
- function<void(int)> dfs2 = [&](int u)
- {
- was[u] = 1;
- rdp[u] = 1;
- for (int to : gt[u])
- {
- if (!was[to])
- dfs2(to);
- rdp[u] += rdp[to];
- }
- };
- for (int i = 0; i < n; ++i)
- {
- if (!was[i])
- dfs2(i);
- }
- int q;
- cin >> q;
- while (q--)
- {
- int i;
- cin >> i;
- --i;
- cout << 1ll * rdp[e[i].X] * dp[e[i].Y] << "\n";
- }
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement