[关闭]
@ysner 2018-08-03T01:01:00.000000Z 字数 3218 阅读 1982

qtree4

点分治


题面

给出一棵边带权的节点数量为的树,初始树上所有节点都是白色。有两种操作:

解析

打这道题的过程中,了许多新姿势。

点分治经典问题。每次分治时找到每棵子树深度最大的一个白点,用最大的两个统计一下即可。
找最大和次大自然需要堆,且每个点都需要开两个。
第一个堆插入这个重心管辖的一坨树所有白点到分治树上这个点父亲的距离。
第二个堆插入所有点分治树上孩子的堆顶,这样我们就可以对于每个分治重心,找到分属两棵子树的深度最大的两个白点,即这个点堆的最大和次大值。

修改就只要再点分树上从自己的重心往根走,然后维护一下那些堆。

注意这些堆要可删除(即可以指定删除其中一个元素)。
原理是每个堆里设一个删除堆和插入堆,插入元素就插入插入堆,删除元素就插入删除堆。如要取堆顶,就把两堆同时弹到堆顶不相等即可,取插入堆堆顶即可。

  1. #include<iostream>
  2. #include<cmath>
  3. #include<cstdio>
  4. #include<cstdlib>
  5. #include<cstring>
  6. #include<algorithm>
  7. #include<queue>
  8. #define ll long long
  9. #define re register
  10. #define il inline
  11. #define inf 1000000000
  12. #define fp(i,a,b) for(re int i=a;i<=b;i++)
  13. #define fq(i,a,b) for(re int i=a;i>=b;i--)
  14. using namespace std;
  15. const int N=5e5+100;
  16. struct Edge{int to,nxt,w;}e[N<<1];
  17. struct dui
  18. {
  19. priority_queue<int>s,t;int sz;
  20. il void pop(re int x){t.push(x);--sz;}
  21. il void push(re int x){s.push(x);++sz;}
  22. il void pre(){while(t.size()&&s.top()==t.top()) s.pop(),t.pop();}
  23. il int top(){pre();return sz?s.top():-inf;}
  24. il int len()
  25. {
  26. if(sz<2) return 0;
  27. re int x=top();pop(x);
  28. re int y=top();push(x);
  29. return max(x+y,0);
  30. }
  31. il void del(re int x,re int k){k?pop(x):push(x);}
  32. }a[N],b[N],ans;
  33. int n,h[N],cnt,Q,d[N],dis[N],f[N],sz[N],mx,mn,Sz[N],top[N],son[N],root,Size,F[N],w[N],tot,s[N],tim;
  34. bool vis[N];
  35. il void add(re int u,re int v,re int w){e[++cnt]=(Edge){v,h[u],w};h[u]=cnt;}
  36. il ll gi()
  37. {
  38. re ll x=0,t=1;
  39. re char ch=getchar();
  40. while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
  41. if(ch=='-') t=-1,ch=getchar();
  42. while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
  43. return x*t;
  44. }
  45. il void dfs1(re int u,re int fa)
  46. {
  47. d[u]=d[fa]+1;f[u]=fa;sz[u]=1;
  48. for(re int i=h[u];i+1;i=e[i].nxt)
  49. {
  50. re int v=e[i].to;
  51. if(v==fa) continue;
  52. dis[v]=dis[u]+e[i].w;
  53. dfs1(v,u);
  54. sz[u]+=sz[v];
  55. if(sz[son[u]]<sz[v]) son[u]=v;
  56. }
  57. }
  58. il void dfs2(re int u,re int up)
  59. {
  60. top[u]=up;
  61. if(son[u]) dfs2(son[u],up);
  62. for(re int i=h[u];i+1;i=e[i].nxt)
  63. {
  64. re int v=e[i].to;
  65. if(v==f[u]||v==son[u]) continue;
  66. dfs2(v,v);
  67. }
  68. }
  69. il int lca(re int u,re int v)
  70. {
  71. while(top[u]^top[v])
  72. {
  73. if(d[top[u]]<d[top[v]]) swap(u,v);
  74. u=f[top[u]];
  75. }
  76. return d[u]<d[v]?u:v;
  77. }
  78. il int Dis(re int u,re int v){return dis[u]+dis[v]-2*dis[lca(u,v)];}
  79. il void Getrt(re int u,re int fa)
  80. {
  81. re int mx=0;Sz[u]=1;s[++tim]=u;
  82. for(re int i=h[u];i+1;i=e[i].nxt)
  83. {
  84. re int v=e[i].to;
  85. if(v==fa||vis[v]) continue;
  86. Getrt(v,u);
  87. Sz[u]+=Sz[v];
  88. mx=max(mx,Sz[v]);
  89. }
  90. mx=max(mx,Size-Sz[u]);
  91. if(mx<mn) mn=mx,root=u;
  92. }
  93. il int dfs(re int u,re int fa)
  94. {
  95. vis[u]=1;F[u]=fa;b[u].push(0);
  96. if(F[u]) fp(i,1,tim) a[u].push(Dis(s[i],F[u]));
  97. for(re int i=h[u];i+1;i=e[i].nxt)
  98. {
  99. re int v=e[i].to;
  100. if(vis[v]) continue;
  101. Size=Sz[v];mn=n;tim=0;
  102. Getrt(v,u);
  103. re int p=dfs(root,u);
  104. b[u].push(a[p].top());
  105. }
  106. ans.push(b[u].len());
  107. return u;
  108. }
  109. il void Modify(re int u,re int w)
  110. {
  111. re int l1=b[u].len(),l2,s1,s2;
  112. b[u].del(0,w);l2=b[u].len();
  113. if(l1^l2) ans.pop(l1),ans.push(l2);
  114. for(re int i=u;F[i];i=F[i])
  115. {
  116. s1=a[i].top();
  117. a[i].del(Dis(u,F[i]),w);
  118. s2=a[i].top();
  119. if(s1==s2) continue;
  120. l1=b[F[i]].len();
  121. if(s1!=-inf) b[F[i]].pop(s1);if(s2!=-inf) b[F[i]].push(s2);
  122. l2=b[F[i]].len();
  123. if(l1^l2) ans.pop(l1),ans.push(l2);
  124. }
  125. }
  126. int main()
  127. {
  128. memset(h,-1,sizeof(h));
  129. tot=n=gi();
  130. fp(i,1,n-1)
  131. {
  132. re int u=gi(),v=gi(),w=gi();
  133. add(u,v,w);add(v,u,w);
  134. }
  135. dfs1(1,0);dfs2(1,1);
  136. Size=mn=n;Getrt(1,0);
  137. dfs(root,0);
  138. fp(i,1,n) w[i]=1;
  139. Q=gi();
  140. while(Q--)
  141. {
  142. re char op='0';while(op!='A'&&op!='C') op=getchar();
  143. if(op=='A')
  144. {
  145. tot?printf("%d\n",tot==1?0:ans.top()):puts("They have disappeared.");
  146. }
  147. if(op=='C')
  148. {
  149. re int u=gi();tot+=(w[u]?-1:1);
  150. Modify(u,w[u]);w[u]^=1;
  151. }
  152. }
  153. return 0;
  154. }
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注