当前位置:   article > 正文

CSP - J day7

CSP - J day7

 一. 砍树

(1)思路:

· 答案具有单调性:对于正确答案 ans ,小于 ans 的都一定符合条件,大于 ans 的一定不符合条件,因此可以二分答案。

· dfs到叶节点,再逐步递归出来,过程中节点数增加,一旦达到 mid ,ans 加一,同时向上贡献节点数变为零。

(2)代码:

  1. #include<bits/stdc++.h>
  2. #define LL long long
  3. using namespace std;
  4. const int N = 1e5 + 5;
  5. int n, k, ans;
  6. vector<int> G[N];
  7. int dfs(int u, int fa, int mid)
  8. {
  9. int s = 1;
  10. for (auto v : G[u])
  11. {
  12. if (v == fa)
  13. continue;
  14. s += dfs(v, u, mid);
  15. }
  16. if (s >= mid)
  17. {
  18. ans ++;
  19. return 0; // µ±Ç°½Úµã¿ªÊ¼µÄ×ÓÊ÷²»»áÔÙ¶ÔÉÏÃæ×ö³ö¹±Ï×
  20. }
  21. return s; // µ±Ç°½ÚµãÄܹ»ÏòÉÏ×ö³ö¹±Ï×
  22. }
  23. bool check(int mid)
  24. {
  25. ans = 0; // ans ²»Êdzõʼ»¯Îª 1 ²¢²»Äܹ»±£Ö¤ËùÓнڵãÊý±¾Éí¾Í´óÓÚ mid
  26. dfs(1, -1, mid);
  27. return ans >= k + 1;
  28. }
  29. int main()
  30. {
  31. cin >> n >> k;
  32. for (int i = 1; i < n; i ++)
  33. {
  34. int u, v;
  35. cin >> u >> v;
  36. G[u].push_back(v), G[v].push_back(u);
  37. }
  38. int l = 1, r = n + 1, mid;
  39. while (l + 1 < r)
  40. {
  41. mid = l + (r - l) / 2;
  42. if (check(mid))
  43. l = mid;
  44. else
  45. r = mid;
  46. }
  47. cout << l;
  48. return 0;
  49. }

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/942136
推荐阅读
相关标签
  

闽ICP备14008679号