이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int inf = 1e9;
const int N = 100'010;
int n, m, q;
vector<pair<int,int>> e;
vector<pair<int,int>> vals;
vector<int> g[N];
bool state[N];
int dfs(int at, int mx = inf, int p = -1) {
// cout << at << " " << mx << endl;
int ans=1;
for(int ed:g[at]) {
int to = (e[ed].first^e[ed].second^at);
if(to == p)continue;
if(vals[ed].first <= mx) {
ans+=dfs(to, min(mx, vals[ed].second), at);
}
}
return ans;
}
int main () {
cin.tie(0)->sync_with_stdio(0);
cin >> n >> m >> q;
for(int i = 1;i<n;i++) {
int u, v;
cin >> u >> v;
e.push_back({u, v});
vals.push_back({inf,-inf});
}
for(int i = 0;i<m;i++) {
int c;
cin >> c;
c--;
state[c]^=1;
if(state[c] == 1 and vals[c].first == inf) {
vals[c].first = i;
}
else vals[c].second = i;
}
for(int i = 0;i<n-1;i++) {
if(vals[i].first == inf)continue;
if(vals[i].second == -inf)vals[i].second = inf;
g[e[i].first].push_back(i);
g[e[i].second].push_back(i);
}
while(q--) {
int a;
cin >> a;
// dfs(a);
cout << dfs(a) << "\n";
// cout << endl;
}
}
| # | 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... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |