이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 5;
const int MAXS = 200;
vector<int> adj[MAXN];
vector<pair<int, int>> caras[MAXN];
int dp[MAXN];
bool mark[MAXN], blocked[MAXN], valid[MAXN];
int solve(int v) {
if (dp[v] != -1) return dp[v];
int& ans = dp[v];
ans = 0;
valid[v] = !blocked[v];
for (int u : adj[v]) {
ans = max(ans, solve(u) + valid[u]);
valid[v] |= valid[u];
}
return ans;
}
void merge(int a, int b) {
vector<pair<int, int>> result;
result.reserve(500);
int j = 0;
for (int i = 0; i < (int) caras[a].size(); i++) {
while (j < (int) caras[b].size() && caras[b][j].first + 1 > caras[a][i].first) {
result.push_back({caras[b][j].first + 1, caras[b][j].second});
j++;
}
result.push_back(caras[a][i]);
}
for (; j < (int) caras[b].size(); j++) {
result.push_back({caras[b][j].first + 1, caras[b][j].second});
}
caras[a].clear();
for (pair<int, int> x : result) {
if (!mark[x.second] && (int) caras[a].size() < MAXS + 1) {
caras[a].push_back(x);
mark[x.second] = true;
}
}
for (pair<int, int> x : caras[a]) {
mark[x.second] = false;
}
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
int n, m, q;
cin >> n >> m >> q;
for (int i = 0; i < m; i++) {
int a, b;
cin >> a >> b;
adj[b].push_back(a);
}
for (int i = 1; i <= n; i++) {
caras[i].push_back({0, i});
for (int u : adj[i]) {
merge(i, u);
}
}
for (int i = 0; i < q; i++) {
int t, y;
cin >> t >> y;
vector<int> bosta;
for (int j = 0; j < y; j++) {
int x;
cin >> x;
blocked[x] = true;
bosta.push_back(x);
}
if (y > MAXS) {
for (int j = 1; j <= n; j++) {
dp[j] = -1;
}
int ans = solve(t);
if (ans == 0 && blocked[t]) ans = -1;
cout << ans << '\n';
} else {
int ans = -1;
for (auto x : caras[t]) {
if (!blocked[x.second]) {
ans = x.first;
break;
}
}
cout << ans << '\n';
}
for (int j : bosta) {
blocked[j] = false;
}
}
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... |