【分治】动态点分治 ([ZJOI2007]捉迷藏)

动态点分治

先看一道题目 [ZJOI2007]捉迷藏

显然如果不带修改O(N)的树形动规和O(NlogN)的静态点分治都可以切掉这道题

一、点分树

考虑点分治,对于每一个分治区域树的重心的答案只会与其所有子区域树有关,所以我们可以再构建一颗点分树:

在点分治的过程中,我们把每个区域树的重心和其子区域树的重心建立父子关系,形成了一颗新的树,称为点分树。

点分树的性质:1、一颗点分树的深度为严格logN的

                         2、当修改一个结点事,只会对其父亲结点造成影响z

二、维护结点信息

对于这道题,我们只需维护每个结点上的最大链和次大链,我们可以对于每个结点建两个大根堆:

A:关键字:原分治区域树中以该结点为链的一段的所有链的长度

B:关键字:原分治树中以该结点为根时,每个子树中最长链的长度

所以我们可以得出每个分支区域树中,B中的第一大元素和第二大元素的和即是答案

B由该结点为根的子树中每个A的最大值组成

最后我们只需用树链剖分求lca,并用差分求树上任意两点距离即可

三、时间复杂度&空间复杂度

时间复杂度为点分治的NlogN+堆的logN == O(Nlog2N)

空间复杂度:O(N)

  1 #include<bits/stdc++.h>
  2 #define INF 0x3f3f3f3f
  3 #define MAXN 100010
  4 using namespace std;
  5 inline int read ()
  6 {
  7     int w=1,s=0;
  8     char ch=getchar ();
  9     while (ch<'0'||ch>'9'){if (ch=='-') w=-1;ch=getchar ();}
 10     while ('0'<=ch&&ch<='9') s=(s<<1)+(s<<3)+(ch^48),ch=getchar ();
 11     return s*w;
 12 }
 13 struct Heap{
 14     priority_queue<int>A,B;int size;
 15     Heap(){size=0;}
 16     void push (int x){A.push (x),size++;}
 17     void pop (int x){B.push (x),size--;}
 18     int top (){while (!B.empty ()&&A.top ()==B.top ()) A.pop (),B.pop ();return size?A.top ():-INF;}
 19     int len ()
 20     {
 21         if (size==0) return -1;
 22         if (size==1) return 0;
 23         int x=top ();pop (x);int y=top ();push (x);
 24         return x+y;
 25     }
 26     void op (int x,int c){c?pop (x):push (x);}
 27 }a[MAXN],b[MAXN],ans;
 28 struct edge{
 29     int v,w,nxt;
 30 }e[MAXN<<1];
 31 int n,q,cnt,root,sum,tot;
 32 int head[MAXN],size[MAXN],maxp[MAXN],s[MAXN],col[MAXN];
 33 int Fa[MAXN],dep[MAXN],son[MAXN],dist[MAXN],top[MAXN];
 34 bool used[MAXN];
 35 char opt[MAXN];
 36 void add (int u,int v,int w)
 37 {
 38     e[++cnt].v=v,e[cnt].w=w,e[cnt].nxt=head[u],head[u]=cnt;
 39 }
 40 void dfs1 (int u,int fa)
 41 {
 42     Fa[u]=fa,size[u]=1,dep[u]=dep[fa]+1;
 43     for (int i=head[u];i!=0;i=e[i].nxt)
 44         if (e[i].v!=fa)
 45         {
 46             dist[e[i].v]=dist[u]+e[i].w;
 47             dfs1 (e[i].v,u);
 48             size[u]+=size[e[i].v];
 49             if (size[e[i].v]>size[son[u]]) son[u]=e[i].v;
 50         }
 51 }
 52 void dfs2 (int u,int topf)
 53 {
 54     top[u]=topf;
 55     if (son[u]) dfs2 (son[u],topf);
 56     for (int i=head[u];i!=0;i=e[i].nxt)
 57         if (e[i].v!=Fa[u]&&e[i].v!=son[u])
 58             dfs2 (e[i].v,e[i].v);
 59 }
 60 int lca (int x,int y)
 61 {
 62     while (dep[top[x]]!=dep[top[y]])
 63     {
 64         if (dep[top[x]]<dep[top[y]]) swap (x,y);
 65         x=Fa[top[x]];
 66     }
 67     return dep[x]<dep[y]?x:y;
 68 }
 69 void getrt (int u,int fa)
 70 {
 71     size[u]=1,maxp[u]=0;
 72     if (!col[u]) s[++tot]=u;
 73     for (int i=head[u];i!=0;i=e[i].nxt)
 74         if (e[i].v!=fa&&!used[e[i].v])
 75         {
 76             getrt (e[i].v,u);
 77             size[u]+=size[e[i].v];
 78             maxp[u]=max (maxp[u],size[e[i].v]);
 79         }
 80     maxp[u]=max (maxp[u],sum-size[u]);
 81     if (maxp[u]<maxp[root]) root=u;
 82 }
 83 int f[MAXN];
 84 int getdis (int x,int y)
 85 {
 86     return dist[x]+dist[y]-(dist[lca (x,y)]<<1);
 87 }
 88 int divide (int u)
 89 {
 90     used[u]=1;if (!col[u]) b[u].push (0);
 91     for (int i=1;i<=tot;i++) a[u].push (getdis (f[u],s[i]));
 92     for (int i=head[u];i!=0;i=e[i].nxt)
 93         if (!used[e[i].v])
 94         {
 95             maxp[root=tot=0]=sum=size[e[i].v],getrt (e[i].v,0);
 96             f[root]=u;
 97             int child=divide (root);
 98             b[u].push (a[child].top ());
 99         }
100     ans.push (b[u].len ());
101     return u;
102 }
103 void modify (int x,int c)
104 {
105     int l1=b[x].len (),l2,s1,s2;b[x].op (0,c);
106     if (l1!=(l2=b[x].len ())) ans.pop (l1),ans.push (l2);
107     for (int u=x,fa=f[u];fa;fa=f[u=fa])
108     {
109         s1=a[u].top (),a[u].op (getdis (fa,x),c),s2=a[u].top();
110         if (s1!=s2)
111         {
112             l1=b[fa].len ();
113             if (s1!=-INF) b[fa].pop (s1);
114             if (s2!=-INF) b[fa].push (s2);
115             l2=b[fa].len ();
116             if (l1!=l2) ans.pop (l1),ans.push (l2);
117         }
118     }
119 }
120 int main()
121 {
122     n=read ();
123     for (int i=1;i<n;i++)
124     {
125         int u=read (),v=read ();
126         add (u,v,1);add (v,u,1);
127     }
128     dfs1 (1,0);dfs2 (1,1);
129     maxp[root]=sum=n,getrt (1,0);
130     divide (root);
131     q=read ();
132     while (q--)
133     {
134         scanf ("%s",opt+1);
135         if (opt[1]=='C')
136         {
137             int x=read ();col[x]^=1;
138             modify (x,col[x]);
139         }
140         else printf ("%d\n",ans.top ()); 
141     }
142     return 0;
143 }

猜你喜欢

转载自www.cnblogs.com/PaulShi/p/10057768.html