const int LG = 21;
const int N = 200005;
const long long MOD = 1e9 + 7;
const long long INF = 1e9;
const long long INFLL = 1e18;
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;
typedef vector<vector<int> > vvi;
typedef vector<ll> vll;
#define forn(i, n) for (int (i) = 0; (i) != (n); (i)++)
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
#define popcount(x) __builtin_popcount(x)
#define popcountll(x) __builtin_popcountll(x)
#define fi first
#define se second
#define re return
#define pb push_back
#define uniq(x) sort(all(x)); (x).resize(unique(all(x)) - (x).begin())
#ifdef LOCAL
#define dbg(x) cerr << __LINE__ << " " << #x << " " << x << endl
#define ln cerr << __LINE__ << endl
#else
#define dbg(x) void(0)
#define ln void(0)
#endif // LOCAL
int dfs(int v, int p, int t, vvi &graph, vi &x, vi &y, vector<vector<pii> > &act)
{
int res = 1;
for (auto i : graph[v])
{
int u = x[i] + y[i] - v;
if (u != p)
{
int L = -1, R = act[i].size();
while (L + 1 < R)
{
int M = (L + R) / 2;
if (act[i][M].first <= t)
R = M;
else
L = M;
}
if (R == act[i].size()) continue;
int T = min(act[i][R].second, t);
res += dfs(u, v, T, graph, x, y, act);
}
}
return res;
}
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n, m, q;
cin >> n >> m >> q;
vector<int> x(n - 1), y(n - 1);
vector<vector<int> > graph(n);
for (int i = 0; i < n - 1; i++)
{
cin >> x[i] >> y[i];
x[i]--, y[i]--;
graph[x[i]].push_back(i);
graph[y[i]].push_back(i);
}
vector<int> fr(n - 1, -1);
vector<vector<pii> > act(n - 1);
for (int i = 0; i < m; i++)
{
int z;
cin >> z;
z--;
if (fr[z] == -1)
{
fr[z] = i;
}
else
{
act[z].push_back({fr[z], i});
fr[z] = -1;
}
}
for (int i = 0; i < n - 1; i++)
{
if (fr[i] != -1)
{
act[i].push_back({fr[i], m});
}
}
for (int i = 0; i < q; i++)
{
int v;
cin >> v;
v--;
cout << dfs(v, v, m, graph, x, y, act) << "\n";
}
}
Compilation message
synchronization.cpp: In function 'int dfs(int, int, int, vvi&, vi&, vi&, std::vector<std::vector<std::pair<int, int> > >&)':
synchronization.cpp:54:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if (R == act[i].size()) continue;
~~^~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Incorrect |
0 ms |
384 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
69 ms |
21364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
384 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
105 ms |
15244 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Incorrect |
0 ms |
384 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |