Codeforces Round #329 (Div. 2) D. Happy Tree Party

Myth丶恋晨 2022-08-10 00:53 123阅读 0赞

对于从a→b 的路径,如果边权值为1,则值不变,否则就是1≤边权值,这样的话最多除64次就为0也就不需要继续除下去了,所以问题的关键在于去压缩哪些连续的边权为1的路径,这里自然就想到并查集啦,具体实现看代码吧!
另:因为更改后的边权值只会变小,所以只会涉及到集合并和查的操作。这一点要想懂。

  1. // whn6325689
  2. // Mr.Phoebe
  3. // http://blog.csdn.net/u013007900
  4. #include <algorithm>
  5. #include <iostream>
  6. #include <iomanip>
  7. #include <cstring>
  8. #include <climits>
  9. #include <complex>
  10. #include <fstream>
  11. #include <cassert>
  12. #include <cstdio>
  13. #include <bitset>
  14. #include <vector>
  15. #include <deque>
  16. #include <queue>
  17. #include <stack>
  18. #include <ctime>
  19. #include <set>
  20. #include <map>
  21. #include <cmath>
  22. #include <functional>
  23. #include <numeric>
  24. #pragma comment(linker, "/STACK:1024000000,1024000000")
  25. using namespace std;
  26. #define eps 1e-9
  27. #define PI acos(-1.0)
  28. #define INF 0x3f3f3f3f
  29. #define LLINF 1LL<<50
  30. #define speed std::ios::sync_with_stdio(false);
  31. typedef long long ll;
  32. typedef unsigned long long ull;
  33. typedef long double ld;
  34. typedef pair<ll, ll> pll;
  35. typedef complex<ld> point;
  36. typedef pair<int, int> pii;
  37. typedef pair<pii, int> piii;
  38. typedef vector<int> vi;
  39. #define CLR(x,y) memset(x,y,sizeof(x))
  40. #define CPY(x,y) memcpy(x,y,sizeof(x))
  41. #define clr(a,x,size) memset(a,x,sizeof(a[0])*(size))
  42. #define cpy(a,x,size) memcpy(a,x,sizeof(a[0])*(size))
  43. #define debug(a) cout << #a" = " << (a) << endl;
  44. #define debugarry(a, n) for (int i = 0; i < (n); i++) { cout << #a"[" << i << "] = " << (a)[i] << endl; }
  45. #define mp(x,y) make_pair(x,y)
  46. #define pb(x) push_back(x)
  47. #define lowbit(x) (x&(-x))
  48. #define MID(x,y) (x+((y-x)>>1))
  49. #define getidx(l,r) (l+r | l!=r)
  50. #define ls getidx(l,mid)
  51. #define rs getidx(mid+1,r)
  52. #define lson l,mid
  53. #define rson mid+1,r
  54. template<class T>
  55. inline bool read(T &n)
  56. {
  57. T x = 0, tmp = 1;
  58. char c = getchar();
  59. while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
  60. if(c == EOF) return false;
  61. if(c == '-') c = getchar(), tmp = -1;
  62. while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
  63. n = x*tmp;
  64. return true;
  65. }
  66. template <class T>
  67. inline void write(T n)
  68. {
  69. if(n < 0)
  70. {
  71. putchar('-');
  72. n = -n;
  73. }
  74. int len = 0,data[20];
  75. while(n)
  76. {
  77. data[len++] = n%10;
  78. n /= 10;
  79. }
  80. if(!len) data[len++] = 0;
  81. while(len--) putchar(data[len]+48);
  82. }
  83. //-----------------------------------
  84. const int MAXN = 10+2e5;
  85. int n, m;
  86. int dep[MAXN], st[MAXN];
  87. ll w[MAXN];
  88. vector<pii> g[MAXN];
  89. pii pre[MAXN];
  90. void dfs(int u,int fa)
  91. {
  92. dep[u] = dep[fa]+1;
  93. for(auto it:g[u])
  94. {
  95. int v = it.first;
  96. int id = it.second;
  97. if(v == fa) continue;
  98. pre[v] = mp(u,id);
  99. dfs(v,u);
  100. }
  101. }
  102. int find_set(int u)
  103. {
  104. if(w[pre[u].second] != 1) return u;
  105. if(st[u] == u)
  106. return st[u] = find_set(pre[u].first);
  107. return st[u] = find_set(st[u]);
  108. }
  109. int jmp(int a,ll &v)
  110. {
  111. int id = pre[a].second;
  112. if(w[id] > 1)
  113. {
  114. v /= w[id];
  115. return pre[a].first;
  116. }
  117. return st[a] = find_set(st[a]);
  118. }
  119. ll cal(int a,int b,ll v)
  120. {
  121. while(v && a!=b)
  122. {
  123. if(dep[a] < dep[b]) swap(a,b);
  124. a = jmp(a,v);
  125. }
  126. return v;
  127. }
  128. int main()
  129. {
  130. // freopen("in.txt","r",stdin);
  131. scanf("%d%d",&n, &m);
  132. for(int i = 1; i < n; i ++)
  133. {
  134. int u,v;
  135. scanf("%d%d%I64d",&u,&v,&w[i]);
  136. g[u].pb(mp(v,i)), g[v].pb(mp(u,i));
  137. }
  138. dfs(1,0);
  139. for(int i = 1; i <= n; i ++) st[i] = i;
  140. while(m --)
  141. {
  142. int tp;
  143. scanf("%d",&tp);
  144. if(tp == 1)
  145. {
  146. int a, b;
  147. ll y;
  148. scanf("%d%d%I64d",&a, &b, &y);
  149. printf("%I64d\n",cal(a,b,y));
  150. }
  151. else
  152. {
  153. int p;
  154. ll c;
  155. scanf("%d%I64d",&p, &c);
  156. w[p] = c;
  157. }
  158. }
  159. return 0;
  160. }

发表评论

表情:
评论列表 (有 0 条评论,123人围观)

还没有评论,来说两句吧...

相关阅读

    相关 codeforces-round375-div2

    题目链接:[codeforces round 375 div2][] A题: 读一遍题目就知道是超级水的题目,,就是给你三个坐标,求三个坐标汇集到一个点所走的路程