Advertisement
pb_jiang

CF337D WA

Mar 7th, 2023
135
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.39 KB | None | 0 0
  1. // Problem: D. Book of Evil
  2. // Contest: Codeforces - Codeforces Round 196 (Div. 2)
  3. // URL: https://codeforces.com/problemset/problem/337/D
  4. // Memory Limit: 256 MB
  5. // Time Limit: 2000 ms
  6. //
  7. // Powered by CP Editor (https://cpeditor.org)
  8.  
  9. #include <assert.h>
  10. #include <bits/stdc++.h>
  11. using namespace std;
  12. #define dbg(...) logger(#__VA_ARGS__, __VA_ARGS__)
  13. template <typename... Args> void logger(string vars, Args &&... values)
  14. {
  15.     cerr << vars << " = ";
  16.     string delim = "";
  17.     (..., (cerr << delim << values, delim = ", "));
  18.     cerr << endl;
  19. }
  20.  
  21. template <class T> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m)); }
  22. template <class T> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n)); }
  23. template <class T, T init> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m, init)); }
  24. template <class T, T init> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n, init)); }
  25.  
  26. template <class T> using mpq = priority_queue<T, vector<T>, greater<T>>;
  27.  
  28. using ll = long long;
  29. using pii = pair<int, int>;
  30. using vl = vector<ll>;
  31. using vi = vector<int>;
  32. using a3i = array<int, 3>;
  33.  
  34. int n, m, dist;
  35. vi g[100003];
  36. set<int> gs;
  37. vector<a3i> stat;
  38. int ans = 0;
  39.  
  40. a3i dfs(int u, int fa)
  41. {
  42.     a3i ret{INT_MIN / 2, 0, INT_MIN / 2};
  43.     if (gs.count(u))
  44.         ret[0] = 0, ret[1] = u;
  45.     for (auto v : g[u]) {
  46.         if (v == fa)
  47.             continue;
  48.         a3i sub = dfs(v, u);
  49.         if (sub[0] + 1 > ret[0]) {
  50.             ret[2] = ret[0];
  51.             ret[0] = sub[0] + 1;
  52.             ret[1] = v;
  53.         } else if (sub[0] > ret[2])
  54.             ret[2] = sub[0];
  55.     }
  56.     return stat[u] = ret;
  57. }
  58. void reroot(int u, int fa, int d)
  59. {
  60.     if (d > dist)
  61.         return;
  62.     const a3i &s = stat[u];
  63.     if (s[0] <= dist && d <= dist)
  64.         ans++;
  65.     for (auto v : g[u]) {
  66.         if (v == fa)
  67.             continue;
  68.         if (v == s[1])
  69.             reroot(v, u, max(d, s[2]) + 1);
  70.         else
  71.             reroot(v, u, max(d, s[0]) + 1);
  72.     }
  73. }
  74.  
  75. int main(int argc, char **argv)
  76. {
  77.     cin >> n >> m >> dist;
  78.     int a, b;
  79.     for (int i = 0; i < m; ++i)
  80.         cin >> a, gs.insert(a);
  81.     for (int i = 1; i < n; ++i)
  82.         cin >> a >> b, g[a].push_back(b), g[b].push_back(a);
  83.     stat = vector<a3i>(100003);
  84.     dfs(1, -1);
  85.     reroot(1, -1, INT_MIN / 2);
  86.     cout << ans << endl;
  87.     return 0;
  88. };
  89.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement