点分治&动态点分治小结

(写篇博客证明自己还活着×2)

转载请注明原文地址:http://www.cnblogs.com/LadyLex/p/8006488.html 

有的时候,我们会发现这样一类题:它长得很像一个$O(n)$树规,

但是却很难用单独的数组维护对应的信息,这样我们就有了淀粉质点分治。

通过直接统计($O(nlogn)$)或者加上数据结构(比如树状数组,堆,线段树等等)维护信息($O(nlog^{2}n)$),

我们可以统计之前不好统计的东西

这篇博客将会介绍点分治以及动态点分治的简单应用,希望阅读本文的你能有所收获,那就再好不过了。

一.树规到点分治

我们先来看一道简单的题目……

2152: 聪聪可可

Time Limit: 3 Sec  Memory Limit: 259 MB

Description

聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:由爸爸在纸上画n个“点”,并用n-1条“边”把这n个“点”恰好连通(其实这就是一棵树)。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),如果两个点之间所有边上数的和加起来恰好是3的倍数,则判聪聪赢,否则可可赢。聪聪非常爱思考问题,在每次游戏后都会仔细研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

Input

输入的第1行包含1个正整数n。后面n-1行,每行3个整数x、y、w,表示x号点和y号点之间有一条边,上面的数是w。

Output

以即约分数形式输出这个概率(即“a/b”的形式,其中a和b必须互质。如果概率为1,输出“1/1”)。

Sample Input

5
1 2 1
1 3 2
1 4 1
2 5 3

Sample Output

13/25
【样例说明】
13组点对分别是(1,1) (2,2) (2,3) (2,5) (3,2) (3,3) (3,4) (3,5) (4,3) (4,4) (5,2) (5,3) (5,5)。

【数据规模】
对于100%的数据,n<=20000。
 
拿到本题,一个显然的思路是统计长度为3的倍数的路径个数,再比上总的路径个数即是答案。
那么我们发现,3的倍数%3=0(废话……),因此我们可以设$f[i][0/1/2]$为以i为根的子树,向下伸出的路径中长度%3=0/1/2的路径数。
这样我们在每一个点简单的合并一下(注意去掉同一子树的贡献)即可。
这样的做法显然是$O(n)$的对吧……
现在让我们考虑一个蠢一点的做法:
我们依然考虑每个点作为路径顶点时符合条件的点的个数,那么我们可以考虑它的每个子树,
数一下里面每个路径的条数,然后合起来对吧。
对于这个节点子树中的节点,我们是化作子问题递归处理的。
那么我们可以把树换个形状……换一个顺序来递归处理
由于重心有个性质:到其他点的距离和最小,
那我们可以暴力数一下每种长度0/1/2的路径有多少,再暴力加上,然后对于被这个重心分割出的几个子树分治递归解决……
这样找重心的复杂度呢?我们最多找$log$次重心就会只剩下一个点,因此复杂度为$O(nlogn)$.(因为每次找到一个重心,新的子树最多是原来子树大小的一半)
那么介绍一下我们必须的操作:
首先是初始化以及数组定义。我的习惯是这样的:
1 int size[N],maxs[N],totsize,root;
2 bool vis[N];
3 inline void intn()
4 {
5     maxs[0]=inf,root=0,dfs1(1,0),solve(root);
6 }
其中$size[i]$表示目前以i点为根的子树size大小 $maxs[i]$为i点size最大的儿子的size(这个“儿子”可以是i点在有根树中的父亲)
$totsize$为当前联通块大小 $root$为最终的重心 $vis[i]$表示i点是否已经被选取作为重心。vis保证我们可以“截取”出准确的对应联通块然后是找重心的部分:
 1 inline void dfs1(int rt,int fa)
 2 {
 3     size[rt]=1,maxsize[rt]=0;
 4     for(int i=adj[rt];i;i=s[i].next)
 5         if(s[i].zhong!=fa&&!mark[s[i].zhong])
 6         {
 7             dfs1(s[i].zhong,rt),
 8             size[rt]+=size[s[i].zhong],
 9             maxsize[rt]=max(maxsize[rt],size[s[i].zhong]);
10         }
11     maxsize[rt]=max(maxsize[rt],totsize-maxsize[rt]);
12     if(maxsize[rt]<maxsize[root])root=rt;
13 }
这其实类似一个树规的$O(n)$过程……这样找到最后$root$就是重心然后是solve过程:
 1 inline void solve(int rt)
 2 {
 3     vis[rt]=1;
 4   /*
 5     deal with ans
 6   */
 7     for(int i=adj[rt];i;i=s[i].next)
 8         if(!vis[s[i].zhong])
 9             totsize=size[s[i].zhong],root=0,
10             dfs1(s[i].zhong,0),solve(root);
11 }
我们在每一次确定的重心上进行计算,calc函数的内容随题目而变,但是大概是长这样的……本题的完整代码:
 1 #include <cstdio>
 2 #include <cstring>
 3 #define N 20010
 4 #define inf 0x7fffffff
 5 #define max(a,b) ((a)>(b)?(a):(b))
 6 int n,e,adj[N],root,totsize,size[N],maxsize[N];
 7 struct edge{int zhong,val,next;}s[N<<1];
 8 inline void add(int qi,int zhong,int val)
 9     {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;s[e].val=val;}
10 bool mark[N];
11 inline void dfs1(int rt,int fa)
12 {
13     size[rt]=1,maxsize[rt]=0;
14     for(int i=adj[rt];i;i=s[i].next)
15         if(!mark[s[i].zhong]&&s[i].zhong!=fa)
16             dfs1(s[i].zhong,rt),size[rt]+=size[s[i].zhong],
17             maxsize[rt]=max(maxsize[rt],size[s[i].zhong]);
18     maxsize[rt]=max(maxsize[rt],totsize-size[rt]);
19     if(maxsize[rt]<maxsize[root])root=rt;
20 }
21 int cnt[3],dist[N],ans;
22 inline void dfs2(int rt,int fa)
23 {
24     ++cnt[dist[rt]];
25     for(int i=adj[rt];i;i=s[i].next)
26         if(!mark[s[i].zhong]&&s[i].zhong!=fa)
27             dist[s[i].zhong]=(dist[rt]+s[i].val)%3,
28             dfs2(s[i].zhong,rt);
29 }
30 inline int calc(int rt,int stval)
31 {
32     dist[rt]=stval,cnt[0]=cnt[1]=cnt[2]=0,dfs2(rt,0);
33     return cnt[0]*cnt[0]+2*cnt[1]*cnt[2];
34 }
35 inline void solve(int rt)
36 {
37     mark[rt]=1,ans+=calc(rt,0);
38     for(int i=adj[rt];i;i=s[i].next)
39         if(!mark[s[i].zhong])
40             ans-=calc(s[i].zhong,s[i].val),
41             totsize=size[s[i].zhong],root=0,
42             dfs1(s[i].zhong,0),solve(root);
43 }
44 inline int gcd(int a,int b){return b==0?a:gcd(b,a%b);}
45 int main()
46 {
47     scanf("%d",&n);
48     register int i,a,b,c;
49     for(i=1;i<n;++i)
50         scanf("%d%d%d",&a,&b,&c),add(a,b,c%3),add(b,a,c%3);
51     totsize=n,root=0,maxsize[0]=inf;
52     dfs1(1,0),solve(root);
53     int di=n*n,d=gcd(di,ans);
54     printf("%d/%d\n",ans/d,di/d);
55 }
BZOJ2152
现在这个思路看起来比较蠢对吧……但是如果换一道题,原本的树规做法就行不通了。

2599: [IOI2011]Race 

Time Limit: 70 Sec  Memory Limit: 128 MB
Description给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000
Input第一行 两个整数 n, k第二..n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始)
Output一个整数 表示最小边数量 如果不存在这样的路径 输出-1
Sample Input
4 3
0 1 1
1 2 2
1 3 4
Sample Output
2
看到这道题,原本树规的做法显然行不通了:
如果我们定义$f[i][j]$为以i为根的子树中长度为j的路径最小边数,空间和时间复杂度都会爆炸。
这个时候我们再想想刚才点分治的做法:
我们仅用一个数组$f[j]$表示到每个重心之后子树中长度为j的路径最小边数,
然后到每个重心之后通过dfs遍历去更新对应的f值,然后再用f数组之间相加==k的元素之和更新答案。
这样,我们就用一个$log$的代价,解决了树规解决很困难的问题。时间复杂度$O(nlogn)$.
代码: 
 1 #include <cstdio>
 2 #include <cstring>
 3 using namespace std;
 4 #define N 200010
 5 #define K 1000010
 6 #define LL long long
 7 #define max(a,b) ((a)>(b)?(a):(b))
 8 #define min(a,b) ((a)<(b)?(a):(b))
 9 int n,k,e,adj[N],ans,cnt[K];
10 bool vis[N];
11 int root,tot,size[N],maxs[N];
12 LL dis[N];
13 struct edge{int zhong,val,next;}s[N<<1];
14 inline void add(int qi,int zhong,int val)
15     {s[++e].zhong=zhong;s[e].val=val;s[e].next=adj[qi];adj[qi]=e;}
16 inline void dfs1(int rt,int fa)
17 {
18     size[rt]=1,maxs[rt]=0;
19     for(int i=adj[rt];i;i=s[i].next)
20         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
21             dfs1(s[i].zhong,rt),size[rt]+=size[s[i].zhong],maxs[rt]=max(maxs[rt],size[s[i].zhong]);
22     maxs[rt]=max(maxs[rt],tot-size[rt]);
23     if(maxs[rt]<maxs[root])root=rt;
24 }
25 inline void dfs2(int rt,int fa,int deep)
26 {
27     if(dis[rt]>=0&&dis[rt]<=k)ans=min(ans,deep+cnt[k-dis[rt]]);
28     for(int i=adj[rt];i;i=s[i].next)
29         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
30             dis[s[i].zhong]=dis[rt]+s[i].val,dfs2(s[i].zhong,rt,deep+1);
31 }
32 inline void update(int rt,int fa,int deep,bool opt)
33 {
34     if(dis[rt]>=0&&dis[rt]<=k)
35         if(opt)cnt[dis[rt]]=min(cnt[dis[rt]],deep);
36         else cnt[dis[rt]]=n;
37     for(int i=adj[rt];i;i=s[i].next)
38         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
39             update(s[i].zhong,rt,deep+1,opt);
40 }
41 inline void solve(int rt)
42 {
43     vis[rt]=1;cnt[0]=0;
44     for(int i=adj[rt];i;i=s[i].next)
45         if(!vis[s[i].zhong])
46             dis[s[i].zhong]=s[i].val,dfs2(s[i].zhong,0,1),update(s[i].zhong,0,1,1);
47     for(int i=adj[rt];i;i=s[i].next)
48         if(!vis[s[i].zhong])
49             update(s[i].zhong,0,1,0);
50     for(int i=adj[rt];i;i=s[i].next)
51         if(!vis[s[i].zhong])
52             root=0,tot=size[s[i].zhong],dfs1(s[i].zhong,0),solve(root);
53 }
54 int main()
55 {
56     // freopen("Ark.in","r",stdin);
57     register int i,a,b,c;scanf("%d%d",&n,&k),ans=n;
58     for(i=1;i<n;++i)scanf("%d%d%d",&a,&b,&c),++a,++b,add(a,b,c),add(b,a,c);
59     for(i=1;i<=k;++i)cnt[i]=n;
60     root=0,maxs[0]=n,tot=n,dfs1(1,0),solve(root),
61     printf("%d\n",(ans==n)?-1:ans);
62 }
BZOJ2599

点分治最核心的思想就是这样,找到重心,分别计算。其他的习题还有:

bzoj4016 (建最短路树,再点分)

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <vector>
 4 #include <algorithm>
 5 using namespace std;
 6 #define N 30010
 7 #define inf 0x3fffffff
 8 #define max(a,b) ((a)>(b)?(a):(b))
 9 int n,m,k,e,adj[N],q[N<<4],hd,tl;bool vis[N];
10 struct edge{int to,next,val;}s[N<<1];
11 inline void add(int qi,int to,int val)
12     {s[++e].to=to;s[e].next=adj[qi];adj[qi]=e;s[e].val=val;}
13 int totsize,root,size[N],maxs[N],maxl[N],cnt[N],dis[N],ans=-inf,tot;
14 struct node{int to,val;};vector<node>to[N];
15 inline bool mt(const node &a,const node &b){return a.to<b.to;}
16 inline void dfs0(int rt)
17 {
18     vis[rt]=1;
19     for(int i=0,l=to[rt].size();i<l;++i)
20         if(!vis[to[rt][i].to]&&dis[to[rt][i].to]==dis[rt]+to[rt][i].val)
21             add(rt,to[rt][i].to,to[rt][i].val),add(to[rt][i].to,rt,to[rt][i].val),dfs0(to[rt][i].to);
22 }
23 inline void spfa_and_build()
24 {
25     memset(dis,0x3f,sizeof(dis));
26     q[hd=tl=1]=1,vis[1]=1,dis[1]=0;
27     register int i,x,l;
28     while(hd<=tl)
29         for(x=q[hd++],vis[x]=0,i=0,l=to[x].size();i<l;++i)
30             if(dis[to[x][i].to]>dis[x]+to[x][i].val)
31             {
32                 dis[to[x][i].to]=dis[x]+to[x][i].val;
33                 if(!vis[to[x][i].to])vis[to[x][i].to]=1,q[++tl]=to[x][i].to;
34             }
35     dfs0(1);
36 }
37 inline void dfs1(int rt,int fa)
38 {
39     size[rt]=1,maxs[rt]=0;
40     for(int i=adj[rt];i;i=s[i].next)
41         if(!vis[s[i].to]&&s[i].to!=fa)
42             dfs1(s[i].to,rt),size[rt]+=size[s[i].to],maxs[rt]=max(maxs[rt],size[s[i].to]);
43     maxs[rt]=max(maxs[rt],totsize-size[rt]);
44     if(maxs[rt]<maxs[root])root=rt;
45 }
46 inline void dfs2(int rt,int fa,int dp)
47 {
48     if(maxl[k-dp-1]!=-inf)
49         if(ans<dis[rt]+maxl[k-dp-1])ans=dis[rt]+maxl[k-dp-1],tot=cnt[k-dp-1];
50         else if(ans==dis[rt]+maxl[k-dp-1])tot+=cnt[k-dp-1];
51     if(dp+1<k)for(int i=adj[rt];i;i=s[i].next)
52         if(!vis[s[i].to]&&s[i].to!=fa)
53             dis[s[i].to]=dis[rt]+s[i].val,dfs2(s[i].to,rt,dp+1);
54 }
55 inline void update(int rt,int fa,int dp)
56 {
57     if(maxl[dp]<dis[rt])maxl[dp]=dis[rt],cnt[dp]=1;
58     else if(maxl[dp]==dis[rt])++cnt[dp];
59     if(dp+1<k)for(int i=adj[rt];i;i=s[i].next)
60         if(!vis[s[i].to]&&s[i].to!=fa)
61             update(s[i].to,rt,dp+1);
62 }
63 inline void solve(int rt)
64 {
65     vis[rt]=1,maxl[0]=0,cnt[0]=1;
66     for(int i=adj[rt];i;i=s[i].next)
67         if(!vis[s[i].to])dis[s[i].to]=s[i].val,dfs2(s[i].to,0,1),update(s[i].to,0,1);
68     for(int i=1;i<=k;++i)maxl[i]=-inf,cnt[i]=0;
69     // for(int i=adj[rt];i;i=s[i].next)if(!vis[s[i].to])update(s[i].to,0,1,0);
70     for(int i=adj[rt];i;i=s[i].next)
71         if(!vis[s[i].to]&&size[s[i].to]>=k)root=0,totsize=size[s[i].to],dfs1(s[i].to,0),solve(root);
72 }
73 inline void work_and_print()
74 {
75     memset(vis,0,sizeof(vis)),memset(dis,0,sizeof(dis)),memset(cnt,0,sizeof(cnt));
76     for(int i=1;i<=k;++i)maxl[i]=-inf;
77     totsize=n,root=0,maxs[0]=inf,dfs1(1,0),solve(root);
78     printf("%d %d",ans,tot);
79 }
80 int main()
81 {
82     scanf("%d%d%d",&n,&m,&k);
83     register int i,j,l,a,b,c;
84     for(i=1;i<=m;++i)
85         scanf("%d%d%d",&a,&b,&c),to[a].push_back((node){b,c}),to[b].push_back((node){a,c});
86     for(i=1;i<=n;++i)sort(to[i].begin(),to[i].end(),mt);
87     spfa_and_build(),work_and_print();
88 }
BZOJ4016

bzoj3672 (推式子,点分治+CDQ维护凸壳)

bzoj1758 (01分数规划,卡精卡时……)

二.从点分治到动态点分治

做着做着题,你可能会发现有的题目要修改……这可怎么办呢。
我们肯定不能一次修改跑一次DP或者点分治吧……
那么我们考虑用数组或者数据结构来维护我们需要的信息,每一次修改都维护数据结构中对应的信息。
这其实借用了树规中的思想,我们通过维护这些信息,可以实现点之间$O(1)$或者$O(logn)$的转移。
可是如果我们对原树中的点维护这些数据结构,一条链就可以送我们上天……
但同时我们观察到,如果我们把点分治时每一层的重心之间连边,这就构成了一颗高度为$logn$的新树,我们叫它分治树。
那么我们就用数据结构维护分治树种对应我们需要的信息。
一个常用的思想:维护当前子树中到父亲的某些信息,从而减小查询的复杂度……
这样说可能有些抽象……我们来看道题:

1095: [ZJOI2007]Hide 捉迷藏

Time Limit: 40 Sec  Memory Limit: 256 MB

Description

  捉迷藏 Jiajia和Wind是一对恩爱的夫妻,并且他们有很多孩子。某天,Jiajia、Wind和孩子们决定在家里玩
捉迷藏游戏。他们的家很大且构造很奇特,由N个屋子和N-1条双向走廊组成,这N-1条走廊的分布使得任意两个屋
子都互相可达。游戏是这样进行的,孩子们负责躲藏,Jiajia负责找,而Wind负责操纵这N个屋子的灯。在起初的
时候,所有的灯都没有被打开。每一次,孩子们只会躲藏在没有开灯的房间中,但是为了增加刺激性,孩子们会要
求打开某个房间的电灯或者关闭某个房间的电灯。为了评估某一次游戏的复杂性,Jiajia希望知道可能的最远的两
个孩子的距离(即最远的两个关灯房间的距离)。 我们将以如下形式定义每一种操作: C(hange) i 改变第i个房
间的照明状态,若原来打开,则关闭;若原来关闭,则打开。 G(ame) 开始一次游戏,查询最远的两个关灯房间的
距离。

Input

  第一行包含一个整数N,表示房间的个数,房间将被编号为1,2,3…N的整数。接下来N-1行每行两个整数a, b,
表示房间a与房间b之间有一条走廊相连。接下来一行包含一个整数Q,表示操作次数。接着Q行,每行一个操作,如
上文所示。

Output

  对于每一个操作Game,输出一个非负整数到hide.out,表示最远的两个关灯房间的距离。若只有一个房间是关
着灯的,输出0;若所有房间的灯都开着,输出-1。

Sample Input

8
1 2
2 3
3 4
3 5
3 6
6 7
6 8
7
G
C 1
G
C 2
G
C 1
G

Sample Output

4
3
3
4

HINT

对于100%的数据, N ≤100000, M ≤500000。

 

 那么我们看这个题,如果这个题不用修改的话,$O(n)$的树规和$O(nlogn)$的点分治都可以解决。

想一下,我们需要什么变量呢?需要最长链和次长链拼起来对吧……

那么我们可以用一些数据结构维护每个点在分治树中伸下去与子树中的每个黑点之间链的长度,

这个数据结构我们可以用大根堆,那么最长链就是堆顶了。

但同时,我们意识到最长链和次长链不能来自同一棵分治树下的子树,

因此我们再来一个数据结构,维护这个点每个分治树儿子的堆顶(这就是分治树中“维护父亲/维护孩子”思想的体现,虽然不是很明显),我们还可以用个堆。

那么这个堆的前两个元素之和就是经过这个点的最长链长度,我们对于每个点都可以维护这样一个变量(当然,也可能不存在)。

那么我们再用一个大根堆维护每个节点的答案,那么这个堆的堆顶就是答案了。

每次修改的时候,我们在第一类堆中删除或者添加对应的链长,维护第二类堆对应子树的元素,再更新第三类堆中的答案,就实现了一次维护。

分治树的深度是$logn$的,堆操作的是$logn$的,因此时间复杂度为$O(nlog^{2}n)$。

代码实现:

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <ctime>
  4 #include <set>
  5 #include <queue>
  6 using namespace std;
  7 #define N 100010
  8 #define inf 0x3fffffff
  9 #define Vt Vater[rt]
 10 int n,e,adj[N];
 11 struct edge{int zhong,next;}s[N<<1];
 12 inline void add(int qi,int zhong)
 13     {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;}
 14 int Vater[N],size[N],root,totsize,maxs[N];
 15 bool state[N],vis[N];
 16 #define max(a,b) ((a)>(b)?(a):(b))
 17 #define min(a,b) ((a)<(b)?(a):(b))
 18 struct heap
 19 {
 20     priority_queue<int>q1,q2;
 21     inline void push(int x){q1.push(x);}
 22     inline void erase(int x){q2.push(x);}
 23     inline int top()
 24     {
 25         while(q2.size()&&q1.top()==q2.top())q1.pop(),q2.pop();
 26         return q1.top();
 27     }
 28     inline void pop()
 29     {
 30         while(q2.size()&&q1.top()==q2.top())q1.pop(),q2.pop();
 31         q1.pop();
 32     }
 33     inline int top2()
 34     {
 35         int val=top();pop();
 36         int ret=top();push(val);
 37         return ret;
 38     }
 39     inline int size()
 40     {
 41         return q1.size()-q2.size();
 42     }
 43 }h1[N],h2[N],h3;
 44 inline void dfs1(int rt,int fa)
 45 {
 46     size[rt]=1,maxs[rt]=0;
 47     for(int i=adj[rt];i;i=s[i].next)
 48         if(s[i].zhong!=fa&&!vis[s[i].zhong])
 49             dfs1(s[i].zhong,rt),size[rt]+=size[s[i].zhong],
 50             maxs[rt]=max(maxs[rt],size[s[i].zhong]);
 51     maxs[rt]=max(maxs[rt],totsize-maxs[rt]);
 52     if(maxs[rt]<maxs[root])root=rt;
 53 }
 54 int f[N][18],bin[25],tp,deep[N];
 55 inline void pre(int rt,int fa)
 56 {
 57     f[rt][0]=fa;deep[rt]=deep[fa]+1;
 58     for(int i=1;bin[i]+1<=deep[rt];++i)f[rt][i]=f[f[rt][i-1]][i-1];
 59     for(int i=adj[rt];i;i=s[i].next)
 60         if(s[i].zhong!=fa)pre(s[i].zhong,rt);
 61 }
 62 inline int LCA(int a,int b)
 63 {
 64     if(deep[a]<deep[b])a^=b,b^=a,a^=b;
 65     int i,cha=deep[a]-deep[b];
 66     for(i=tp;~i;--i)if(cha&bin[i])a=f[a][i];
 67     if(a==b)return a;
 68     for(i=tp;~i;--i)
 69         if(f[a][i]!=f[b][i])a=f[a][i],b=f[b][i];
 70     return f[a][0];
 71 }
 72 inline int dis(int a,int b)
 73     {return deep[a]+deep[b]-(deep[LCA(a,b)]<<1);}
 74 inline void dfs3(int rt,int fa,int Vatty)
 75 {
 76     h1[root].push(dis(rt,Vatty));
 77     for(int i=adj[rt];i;i=s[i].next)
 78         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
 79             dfs3(s[i].zhong,rt,Vatty);
 80 }
 81 inline void dfs2(int rt,int fa)
 82 {
 83     Vt=fa,vis[rt]=1,h2[rt].push(0);
 84     int siz=totsize;
 85     for(int i=adj[rt];i;i=s[i].next)
 86         if(!vis[s[i].zhong])
 87         {
 88             if(size[s[i].zhong]>size[rt])
 89                 totsize=siz-size[rt];
 90             else 
 91                 totsize=size[s[i].zhong];
 92             root=0,dfs1(s[i].zhong,0),dfs3(root,0,rt);
 93             h2[rt].push(h1[root].top()),dfs2(root,rt);
 94         }
 95     if(h2[rt].size()>1)h3.push(h2[rt].top()+h2[rt].top2());
 96 }
 97 inline void turnoff(int who)
 98 {
 99     int val,tmp;
100     if(h2[who].size()>1)h3.erase(h2[who].top()+h2[who].top2());
101     h2[who].push(0);
102     if(h2[who].size()>1)h3.push(h2[who].top()+h2[who].top2());
103     //queue empty() 后依然有数
104     for(int rt=who;Vt;rt=Vt)
105     {
106         if(h2[Vt].size()>1)h3.erase(h2[Vt].top()+h2[Vt].top2());
107         if(h1[rt].size())h2[Vt].erase(h1[rt].top());
108         h1[rt].push(dis(who,Vt));
109         h2[Vt].push(h1[rt].top());
110         if(h2[Vt].size()>1)h3.push(h2[Vt].top()+h2[Vt].top2());
111     }
112 }
113 inline void turnon(int who)
114 {
115     int val,tmp;
116     if(h2[who].size()>1)h3.erase(h2[who].top()+h2[who].top2());
117     h2[who].erase(0);
118     if(h2[who].size()>1)h3.push(h2[who].top()+h2[who].top2());
119     //queue empty()后依然有数
120     for(int rt=who;Vt;rt=Vt)
121     {
122         if(h2[Vt].size()>1)h3.erase(h2[Vt].top()+h2[Vt].top2());
123         h2[Vt].erase(h1[rt].top());
124         h1[rt].erase(dis(who,Vt));
125         if(h1[rt].size())h2[Vt].push(h1[rt].top());
126         if(h2[Vt].size()>1)h3.push(h2[Vt].top()+h2[Vt].top2());
127     }
128 }
129 char B[1<<15],X=0,*S=B,*T=B;
130 #define getc ( S==T&&( T=(S=B)+fread(B,1,1<<15,stdin),S==T )?0:*S++ )
131 inline int read()
132 {
133     int x=0;while(X<'0'||X>'9')X=getc;
134     while(X>='0'&&X<='9')x=10*x+(X^48),X=getc;
135     return x;
136 }
137 inline void readc(){X=getc;while(X<'A'||X>'Z')X=getc;}
138 int main()
139 {
140     // freopen("hide1.in","r",stdin);
141     // freopen("hide.out","w",stdout);
142     n=read();
143     register int i,j,q,a,b,cnt=n;
144     for(bin[0]=i=1;i<=20;++i)bin[i]=bin[i-1]<<1;
145     while(bin[tp+1]<=n)++tp;
146     for(i=1;i<n;++i)
147         a=read(),b=read(),add(a,b),add(b,a);
148     pre(1,0);
149     maxs[0]=inf,root=0,totsize=n,dfs1(1,0),dfs2(root,0);
150     q=read();
151     while(q--)
152     {
153         readc();
154         if(X=='C')
155         {
156             i=read();
157             if(state[i])++cnt,turnoff(i);
158             else --cnt,turnon(i);
159             state[i]^=1;
160         }
161         else
162         {
163             if(cnt<2)printf("%d\n",cnt-1);
164             else printf("%d\n",h3.top());
165         }
166     }
167 }
BZOJ1095

其实这道题是比较简单的……那么我们再看一道题。

3924: [Zjoi2015]幻想乡战略游戏

Time Limit: 100 Sec  Memory Limit: 256 MB

Description

 傲娇少女幽香正在玩一个非常有趣的战略类游戏,本来这个游戏的地图其实还不算太大,幽香还能管得过来,但是不知道为什么现在的网游厂商把游戏的地图越做越大,以至于幽香一眼根本看不过来,更别说和别人打仗了。 在打仗之前,幽香现在面临一个非常基本的管理问题需要解决。 整个地图是一个树结构,一共有n块空地,这些空地被n-1条带权边连接起来,使得每两个点之间有一条唯一的路径将它们连接起来。在游戏中,幽香可能在空地上增加或者减少一些军队。同时,幽香可以在一个空地上放置一个补给站。 如果补给站在点u上,并且空地v上有dv个单位的军队,那么幽香每天就要花费dv×dist(u,v)的金钱来补给这些军队。由于幽香需要补给所有的军队,因此幽香总共就要花费为Sigma(Dv*dist(u,v),其中1<=V<=N)的代价。其中dist(u,v)表示u个v在树上的距离(唯一路径的权和)。 因为游戏的规定,幽香只能选择一个空地作为补给站。在游戏的过程中,幽香可能会在某些空地上制造一些军队,也可能会减少某些空地上的军队,进行了这样的操作以后,出于经济上的考虑,幽香往往可以移动他的补给站从而省一些钱。但是由于这个游戏的地图是在太大了,幽香无法轻易的进行最优的安排,你能帮帮她吗? 你可以假定一开始所有空地上都没有军队。

PDF版试题:JudgeOnline/upload/201708/zjoi2015d1.pdf

Input

第一行两个数n和Q分别表示树的点数和幽香操作的个数,其中点从1到n标号。 
接下来n-1行,每行三个正整数a,b,c,表示a和b之间有一条边权为c的边。 
接下来Q行,每行两个数u,e,表示幽香在点u上放了e单位个军队
(如果e<0,就相当于是幽香在u上减少了|e|单位个军队,说白了就是du←du+e)。
数据保证任何时刻每个点上的军队数量都是非负的。 
1<=c<=1000, 0<=|e|<=1000, n<=10^5, Q<=10^5
对于所有数据,这个树上所有点的度数都不超过20
N,Q>=1
 

Output

 对于幽香的每个操作,输出操作完成以后,每天的最小花费,也即如果幽香选择最优的补给点进行补给时的花费。 

Sample Input

10 5
1 2 1
2 3 1
2 4 1
1 5 1
2 61
2 7 1
5 8 1
7 91
1 10 1
3 1
2 1
8 1
3 1
4 1

Sample Output

0
1
4
5
6
 
怎么操作呢?我们不难发现,由于目标函数是$Σd_{v}*dis(u,v)$,由定义可知,我们的目标其实就是求带权的重心
那么怎么求呢……还是先假设我们有一棵静态的树,那么考虑树规。
想一下树规中我们都求了什么:
设$s1[i]$为i子树点权和,$s2[i]$为i子树点到它的距离和,$s3[i]$为i父亲到它的距离和。
那么转移应该是
$ans[i]=ans[fa]+s2[fa]-s2[i]-s1[i]*dis(i,fa)+s3[fa]+(sum-s1[i])*dis(i,fa)$
dis我们可以用ST表RMQ或者倍增LCA来求.
但其实我们是可以不要s3数组,一层一层跳着往上求$s2[fa]-s2[i]-s1[i]*dis(i,fa)$这个式子的。
由于原树可能是条链,我们不敢直接跳,只好维护s3数组。
但是如果修改的话,s3涉及到的修改会特别多!
我们联想到,分治树的高度是$logn$的!
因此我们可以不维护s3,仅仅维护s1和s2,在分治树上用$O(logn)$的时间求出某一个点作为带权重心时的权和。
但是你会发现,我们上面的式子里有一个-s2[i]很难处理:在树规中i和fa只隔了一条边,但是在分治树中,你并不知道元素长什么样子!
因此我们还是得加一个s3数组,s3[i]表示i的分治树子节点到i分治树父亲的权和。这就是我们动态树分治中“维护父亲”思想的典型体现
那么我们现在就可以以$O(logn)$的复杂度枚举i的祖先fa,求$Σs2[fa]-s3[i]+(s1[fa]-s1[i])*dis(i,fa)$来求与i在原树上的lca是fa的点到i的距离了。
*这种$O(logn)$跳父亲的做法也是一种常见的思想。
然后对于统计答案,我们可以每次从分治树的根出发,看往他的哪个儿子走答案更优。
我们最多走$logn$层,每一层最多进行20次$O(logn)$的计算,因此查询是$O(20log^{2}n)$的.
总的复杂度$O(20Qlog^{2}n)$。代码见下:
  1 #include <cstdio>
  2 #include <cstring>
  3 using namespace std;
  4 #define LL long long
  5 #define N 100010
  6 #define LL long long
  7 #define max(a,b) ((a)>(b)?(a):(b))
  8 #define min(a,b) ((a)<(b)?(a):(b))
  9 struct edge{int zhong,next,val;};
 10 struct G
 11 {
 12     edge s[N<<1];int e,adj[N];
 13     G(){e=0;memset(adj,0,sizeof(adj));}
 14     inline void add(int qi,int zhong,int val=0)
 15         {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;s[e].val=val;}
 16 }T1,T2;
 17 int n,q,f[N<<1][18],logn[N<<1],bin[25],tp;
 18 LL s[3][N],sum,ans,len[N];
 19 int dfn[N],num;
 20 inline void dfs1(int rt,int fa)
 21 {
 22     f[(dfn[rt]=++num)][0]=len[rt];
 23     for(int i=T1.adj[rt];i;i=T1.s[i].next)
 24         if(T1.s[i].zhong!=fa)
 25             len[T1.s[i].zhong]=len[rt]+T1.s[i].val,dfs1(T1.s[i].zhong,rt),f[++num][0]=len[rt];
 26 }
 27 inline LL LCA(int a,int b)
 28 {
 29     if(dfn[a]>dfn[b])a^=b,b^=a,a^=b;
 30     int k=logn[dfn[b]-dfn[a]+1];
 31     return min(f[dfn[a]][k],f[dfn[b]-bin[k]+1][k])<<1;
 32 }
 33 inline LL dis(int a,int b){return len[a]+len[b]-LCA(a,b);}
 34 int size[N],maxs[N],totsize,root,Vater[N];
 35 bool vis[N];
 36 inline void dfs2(int rt,int fa)
 37 {
 38     size[rt]=1,maxs[rt]=0;
 39     for(int i=T1.adj[rt];i;i=T1.s[i].next)
 40         if(!vis[T1.s[i].zhong]&&T1.s[i].zhong!=fa)
 41             dfs2(T1.s[i].zhong,rt),size[rt]+=size[T1.s[i].zhong],
 42             maxs[rt]=max(size[T1.s[i].zhong],maxs[rt]);
 43     maxs[rt]=max(totsize-size[rt],maxs[rt]);
 44     if(maxs[rt]<maxs[root])root=rt;
 45 }
 46 inline void dfs3(int rt,int fa)
 47 {
 48     Vater[rt]=fa;vis[rt]=1;int siz=totsize;
 49     for(int i=T1.adj[rt];i;i=T1.s[i].next)
 50         if(!vis[T1.s[i].zhong])
 51         {
 52             if(size[T1.s[i].zhong]>size[rt])totsize=siz-size[rt];
 53             else totsize=size[T1.s[i].zhong];
 54             root=0,dfs2(T1.s[i].zhong,0),T2.add(rt,root,T1.s[i].zhong),dfs3(root,rt);
 55         }
 56 }
 57 inline void update(int who,LL val)
 58 {
 59     for(int rt=who;rt;rt=Vater[rt])
 60     {
 61         s[0][rt]+=val,s[1][rt]+=dis(rt,who)*val;
 62         if(Vater[rt])s[2][rt]+=dis(Vater[rt],who)*val;
 63     }
 64 }
 65 inline LL calc(int rt)
 66 {
 67     LL ret=0;
 68     for(int x=rt;x;x=Vater[x])
 69     {
 70         ret+=s[1][x];
 71         if(Vater[x])ret+=-s[2][x]+(s[0][Vater[x]]-s[0][x])*dis(Vater[x],rt);
 72     }
 73     return ret;
 74 }
 75 inline LL getans(int rt)
 76 {
 77     LL temp=calc(rt);
 78     for(int i=T2.adj[rt];i;i=T2.s[i].next)
 79         if(calc(T2.s[i].val)<temp)return getans(T2.s[i].zhong);//*******
 80     return temp;
 81 }
 82 char B[1<<15],*S=B,*T=B;
 83 #define getc ( S==T&&(T=(S=B)+fread(B,1,1<<15,stdin),S==T)?0:*S++)
 84 inline int read()
 85 {
 86     int x=0,f=1;register char c=getc;
 87     while(c<'0'||c>'9'){if(c=='-')f=-1;c=getc;}
 88     while(c>='0'&&c<='9')x=10*x+(c^48),c=getc;
 89     return x*f;
 90 }
 91 int main()
 92 {
 93     register int i,j,orig,a,b,c;
 94     n=read();q=read();
 95     for(bin[0]=i=1;i<=20;++i)bin[i]=bin[i-1]<<1;
 96     while(bin[tp+1]<=(n<<1))++tp;
 97     for(logn[0]=-1,i=1;i<=(n<<1);++i)logn[i]=logn[i>>1]+1;
 98     for(i=1;i<n;++i)
 99         a=read(),b=read(),c=read(),T1.add(a,b,c),T1.add(b,a,c);
100     dfs1(1,0),root=0,maxs[0]=n+1,totsize=n,dfs2(1,0);
101     for(i=1;i<=tp;++i)
102         for(j=1;j+bin[i]-1<=(n<<1);++j)
103             f[j][i]=min(f[j][i-1],f[j+bin[i-1]][i-1]);
104     orig=root,dfs3(root,0);
105     while(q--)
106         a=read(),b=read(),update(a,b),printf("%lld\n",getans(orig));
107 }
BZOJ4012

 我们再来一道大同小异的题目:

4012: [HNOI2015]开店

Time Limit: 70 Sec  Memory Limit: 512 MB

Description

 风见幽香有一个好朋友叫八云紫,她们经常一起看星星看月亮从诗词歌赋谈到

人生哲学。最近她们灵机一动,打算在幻想乡开一家小店来做生意赚点钱。这样的
想法当然非常好啦,但是她们也发现她们面临着一个问题,那就是店开在哪里,面
向什么样的人群。很神奇的是,幻想乡的地图是一个树形结构,幻想乡一共有 n
个地方,编号为 1 到 n,被 n-1 条带权的边连接起来。每个地方都住着一个妖怪,
其中第 i 个地方的妖怪年龄是 x_i。妖怪都是些比较喜欢安静的家伙,所以它们并
不希望和很多妖怪相邻。所以这个树所有顶点的度数都小于或等于 3。妖怪和人一
样,兴趣点随着年龄的变化自然就会变化,比如我们的 18 岁少女幽香和八云紫就
比较喜欢可爱的东西。幽香通过研究发现,基本上妖怪的兴趣只跟年龄有关,所以
幽香打算选择一个地方 u(u为编号),然后在 u开一家面向年龄在 L到R 之间(即
年龄大于等于 L、小于等于 R)的妖怪的店。也有可能 u这个地方离这些妖怪比较
远,于是幽香就想要知道所有年龄在 L 到 R 之间的妖怪,到点 u 的距离的和是多
少(妖怪到 u 的距离是该妖怪所在地方到 u 的路径上的边的权之和) ,幽香把这个
称为这个开店方案的方便值。幽香她们还没有决定要把店开在哪里,八云紫倒是准
备了很多方案,于是幽香想要知道,对于每个方案,方便值是多少呢。
 

Input

 第一行三个用空格分开的数 n、Q和A,表示树的大小、开店的方案个数和妖

怪的年龄上限。 
第二行n个用空格分开的数 x_1、x_2、…、x_n,x_i 表示第i 个地点妖怪的年
龄,满足0<=x_i<A。(年龄是可以为 0的,例如刚出生的妖怪的年龄为 0。) 
接下来 n-1 行,每行三个用空格分开的数 a、b、c,表示树上的顶点 a 和 b 之
间有一条权为c(1 <= c <= 1000)的边,a和b 是顶点编号。 
接下来Q行,每行三个用空格分开的数 u、 a、 b。对于这 Q行的每一行,用 a、
b、A计算出 L和R,表示询问“在地方 u开店,面向妖怪的年龄区间为[L,R]的方
案的方便值是多少”。对于其中第 1 行,L 和 R 的计算方法为:L=min(a%A,b%A), 
R=max(a%A,b%A)。对于第 2到第 Q行,假设前一行得到的方便值为 ans,那么当
前行的 L 和 R 计算方法为: L=min((a+ans)%A,(b+ans)%A), 
R=max((a+ans)%A,(b+ans)%A)。 
 

Output

对于每个方案,输出一行表示方便值。 

 

Sample Input

10 10 10
0 0 7 2 1 4 7 7 7 9
1 2 270
2 3 217
1 4 326
2 5 361
4 6 116
3 7 38
1 8 800
6 9 210
7 10 278
8 9 8
2 8 0
9 3 1
8 0 8
4 2 7
9 7 3
4 7 0
2 2 7
3 2 1
2 3 4

Sample Output

1603
957
7161
9466
3232
5223
1879
1669
1282
0

HINT

 满足 n<=150000,Q<=200000。对于所有数据,满足 A<=10^9

 

现在你应该能想到怎么做了吧!

我们用数据结构维护以i为分治树根的子树每个年龄的妖怪到i的距离,以及到i分治树父亲的距离,然后用数据结构求和。

数据结构用什么呢?线段树?$O(nlog^{2}n)$的空间复杂度肯定会MLE的。

注意到这个求和可以写为前缀和相减的形式,因此我们用可以用$O(nlogn)$的空间复杂度的vector+sort排序节省内存。

那么这道题就被解决啦……代码见下:

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <vector>
  4 #include <algorithm>
  5 using namespace std;
  6 #define N 150010
  7 #define LL long long
  8 int n,e,tot,adj[N],q,maxn,val[N];
  9 char B[1<<15],*S=B,*T=B;
 10 #define getc (S==T&&(T=(S=B)+fread(B,1,1<<15,stdin),S==T)?0:*S++)
 11 inline int read()
 12 {
 13     int x=0;register char c=getc;
 14     while(c<'0'||c>'9')c=getc;
 15     while(c>='0'&&c<='9')x=10*x+(c^48),c=getc;
 16     return x;
 17 }
 18 #define min(a,b) ((a)<(b)?(a):(b))
 19 #define max(a,b) ((a)>(b)?(a):(b))
 20 struct edge{int zhong,next,val;}s[N<<1];
 21 inline void add(int qi,int zhong,int val)
 22     {s[++e].zhong=zhong,s[e].val=val,s[e].next=adj[qi],adj[qi]=e;}
 23 int f[N<<1][19],deep[N],dfn[N],num,bin[25],tp,logn[N<<1];
 24 inline void ST()
 25 {
 26     for(int i=1;i<=tp;++i)
 27         for(int j=1;j+bin[i]-1<=(n<<1);++j)
 28             f[j][i]=min(f[j][i-1],f[j+bin[i-1]][i-1]);
 29 }
 30 inline void dfs1(int rt,int fa,int len)
 31 {
 32     f[(dfn[rt]=++num)][0]=deep[rt]=len;
 33     for(int i=adj[rt];i;i=s[i].next)
 34         if(s[i].zhong!=fa)
 35             dfs1(s[i].zhong,rt,len+s[i].val),f[++num][0]=len;
 36 }
 37 int Vater[N],size[N],maxs[N],totsize,root;
 38 bool vis[N];
 39 inline void dfs2(int rt,int fa)
 40 {
 41     size[rt]=1,maxs[rt]=0;
 42     for(int i=adj[rt];i;i=s[i].next)
 43         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
 44             dfs2(s[i].zhong,rt),size[rt]+=size[s[i].zhong],
 45             maxs[rt]=max(maxs[rt],size[s[i].zhong]);
 46     maxs[rt]=max(maxs[rt],totsize-size[rt]);
 47     if(maxs[rt]<maxs[root])root=rt;
 48 }
 49 inline LL dis(int a,int b)
 50 {
 51     if(dfn[a]>dfn[b])a^=b,b^=a,a^=b;
 52     int k=logn[dfn[b]-dfn[a]+1];
 53     return deep[a]+deep[b]-(min(f[dfn[a]][k],f[dfn[b]-bin[k]+1][k])<<1);
 54 }
 55 struct node
 56 {
 57     int val;LL size[3];
 58     node(int a=0,LL b=0,LL c=0,LL d=0){val=a,size[0]=b,size[1]=c,size[2]=d;}
 59     inline bool operator < (const node &b)const
 60         {return val<b.val;}
 61 };
 62 vector<node>sta[N];
 63 inline void dfs3(int rt,int fa,int Vatti)
 64 {
 65     sta[Vatti].push_back(node(val[rt],1,dis(rt,Vatti), Vater[Vatti]?dis(rt,Vater[Vatti]):0 ));
 66     for(int i=adj[rt];i;i=s[i].next)
 67         if(s[i].zhong!=fa&&!vis[s[i].zhong])
 68             dfs3(s[i].zhong,rt,Vatti);
 69 }
 70 inline void dfs4(int rt,int fa)
 71 {
 72     Vater[rt]=fa,vis[rt]=1;
 73     int siz=totsize;
 74     dfs3(rt,0,rt);sta[rt].push_back(node(-1,0,0,0));
 75     sort(sta[rt].begin(),sta[rt].end());
 76     for(int i=0,j=sta[rt].size();i<j-1;++i)
 77         sta[rt][i+1].size[0]+=sta[rt][i].size[0],
 78         sta[rt][i+1].size[1]+=sta[rt][i].size[1],
 79         sta[rt][i+1].size[2]+=sta[rt][i].size[2];
 80     for(int i=adj[rt];i;i=s[i].next)
 81         if(!vis[s[i].zhong])
 82         {
 83             if(size[s[i].zhong]>size[rt])totsize=siz-size[rt];
 84             else totsize=size[s[i].zhong];
 85             root=0,dfs2(s[i].zhong,0),dfs4(root,rt);
 86         }
 87 }
 88 inline node query(int id,int l,int r)
 89 {
 90     if(id==0)return node();
 91     vector<node>::iterator it1=upper_bound(sta[id].begin(),sta[id].end(),node(r,0,0,0) );--it1;
 92     vector<node>::iterator it2=upper_bound(sta[id].begin(),sta[id].end(),node(l-1,0,0,0));--it2;
 93     return node(0,it1->size[0]-it2->size[0],it1->size[1]-it2->size[1],it1->size[2]-it2->size[2]);
 94 }
 95 inline LL calc(int rt,int l,int r)
 96 {
 97     LL ret=0;
 98     // printf("rt=%d l=%d r=%d\n",rt,l,r);
 99     for(int x=rt;x;x=Vater[x])
100     {
101         node a=query(x,l,r);
102         ret+=a.size[1];
103         if(x!=rt)ret+=a.size[0]*dis(x,rt);
104         if(Vater[x])ret-=a.size[2]+a.size[0]*dis(Vater[x],rt);
105     }
106     return ret;
107 }
108 int main()
109 {
110     register int i,j,a,b,c;LL ans=0;
111     n=read(),q=read(),maxn=read();
112     for(bin[0]=i=1;i<=20;++i)bin[i]=bin[i-1]<<1;
113     while(bin[tp+1]<=(n<<1))++tp;
114     for(logn[0]=-1,i=1;i<=(n<<1);++i)logn[i]=logn[i>>1]+1;
115     for(i=1;i<=n;++i)val[i]=read();
116     for(i=1;i<n;++i)
117         a=read(),b=read(),c=read(),add(a,b,c),add(b,a,c);
118     dfs1(1,0,0),ST();
119     root=0,maxs[0]=n+1,totsize=n,dfs2(1,0);
120     dfs4(root,0);
121     while(q--)
122     {
123         a=read(),b=read(),c=read();
124         b=(b+ans)%maxn,c=(c+ans)%maxn;
125         if(b>c)b^=c,c^=b,b^=c;
126         printf("%lld\n",ans=calc(a,b,c));
127     }
128 }
BZOJ4012

最后我们来个大boss试试知难♂而上

 

3435: [Wc2014]紫荆花之恋

Time Limit: 240 Sec  Memory Limit: 512 MB

Description

强强和萌萌是一对好朋友。有一天他们在外面闲逛,突然看到前方有一棵紫荆树。这已经是紫荆花飞舞的季节了,无数的花瓣以肉眼可见的速度从紫荆树上长了出来。仔细看看的话,这个大树实际上是一个带权树。每个时刻它会长出一个新的叶子节点。每个节点上有一个可爱的小精灵,新长出的节点上也会同时出现一个新的小精灵。小精灵是很萌但是也很脆弱的生物,每个小精灵 i 都有一个感受能力值Ri ,小精灵 i, j 成为朋友当且仅当在树上 i 和 j 的距离 dist(i,j) ≤ Ri + R! ,其中 dist(i, j)表示在这个树上从 i 到 j 的唯一路径上所有边的边权和。强强和萌萌很好奇每次新长出一个叶子节点之后,这个树上总共有几对朋友。  
我们假定这个树一开始为空,节点按照加入的顺序从 1开始编号。由于强强非常好奇, 你必须在他每次出现新节点后马上给出总共的朋友对数,不能拖延哦。 

Input

共有 n + 2 行。 
第一行包含一个正整数,表示测试点编号。 
第二行包含一个正整数 n ,表示总共要加入的节点数。 
我们令加入节点前的总共朋友对数是 last_ans,在一开始时它的值为0。 
接下来 n 行中第 i 行有三个数 ai, bi, ri,表示节点  i  的父节点的编号为 ai xor (last_ans mod 10^9)   (其中xor 表示异或,mod  表示取余,数据保证这样操作后得到的结果介于 1到i  –  1之间),与父节点之间的边权为 ci,节点 i 上小精灵的感受能力值为r!。 
注意 a1 = c1 = 0,表示 1 号点是根节点,对于 i > 1,父节点的编号至少为1。

Output

包含 n 行,每行输出1 个整数, 表示加入第 i 个点之后,树上有几对朋友。

Sample Input

0
5
0 0 6
1 2 4
0 9 4
0 5 5
0 2 4 

Sample Output

0
1
2
4

HINT

1<=Ci<=10000
Ai<=2*10^9
Ri<=10^9
N<=100000

 

那么这题怎么做呢……

我们先考虑把题设式子变形:

\[dis(u,v)<=r_{u}+r_{v}  \]

\[dis(u,lca)+dis(lca,v)<=r_{u}+r_{v}  \]

\[r_{u}-dis(lca,u)>=dis(lca,v)-r_{v}\]

那么接下来我们沿用上面几题的思想,在原树上每个节点开2棵平衡树,维护以i为根的子树中的$dis(i,u)-r_{u}$值和$dis(fa[i],u)-r_{u}$值。

这样每次我们从新插入的节点开始往上爬到根,并且更新答案即可。

但是我们发现,如果这是条链……时间复杂度会被卡到$O(n^{2})$

但是我们转念一想,点分树是相对平衡的,深度是$logn$级别的呀!

所以我们借用替罪羊拍扁重建的思想,如果在插入结束后,

某个节点的某个儿子特别大(用$alpha$判断),我们就把那棵子树重构为一棵分治树。

这样我们的复杂度可以有$O(nlog^{2}n)$的保障。

在代码实现的时候,要特别注意重构时和原来这个子树的分治树父亲之间信息的维护。

代码:

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <cstdlib>
  4 #include <vector>
  5 using namespace std;
  6 #define LL long long
  7 #define inf 1000000000
  8 #define N 100010
  9 #define alpha 0.755
 10 int n,e,adj[N],val[N];
 11 struct edge{int zhong,next;}s[N<<1];
 12 inline void add(int qi,int zhong)
 13     {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;}
 14 vector<int> to[N];
 15 int f[N][18],bin[25],tp,deep[N],len[N];
 16 inline int LCA(int a,int b)
 17 {
 18     if(deep[a]<deep[b])a^=b,b^=a,a^=b;
 19     int i,cha=deep[a]-deep[b];
 20     for(i=tp;~i;--i)if(cha&bin[i])a=f[a][i];
 21     if(a==b)return a;
 22     for(i=tp;~i;--i)if(f[a][i]!=f[b][i])a=f[a][i],b=f[b][i];
 23     return f[a][0];
 24 }
 25 inline int dis(int a,int b){return len[a]+len[b]-len[LCA(a,b)]*2;}
 26 struct Goat
 27 {
 28     int val,size;Goat *ch[2];
 29     Goat(){}
 30     inline bool bad()
 31         {return ch[0]->size>=size*alpha+5 || ch[1]->size>=size*alpha+5; }
 32 }*tree1[N],*tree2[N],mem[N<<8],*pool[N<<8],*null,*sta[N];
 33 int tot,top;
 34 inline void intn()
 35 {
 36     null=new Goat();
 37     null->ch[0]=null->ch[1]=null,null->val=null->size=0;
 38     for(int i=0;i<(N<<8);++i)pool[i]=mem+i;
 39     tot=(N<<8)-1;
 40     for(int i=0;i<=n;++i)tree1[i]=tree2[i]=null;
 41 }
 42 inline Goat** insert(Goat *&a,int val)
 43 {
 44     if(a==null)
 45     {
 46         a=pool[tot--],a->ch[0]=a->ch[1]=null;
 47         a->val=val,a->size=1;return &null;
 48     }
 49     ++a->size;
 50     Goat **o=insert(a->ch[a->val<val],val);
 51     if(a->bad())o=&a;return o;
 52 }
 53 inline int get_rank(Goat *o,int val)
 54 {
 55     if(o==null)return 0;
 56     return (o->val>=val)?get_rank(o->ch[0],val):(get_rank(o->ch[1],val)+o->ch[0]->size+1);
 57 }
 58 inline void Erholung(Goat *o)
 59 {
 60     if(o==null)return;
 61     if(o->ch[0]!=null)Erholung(o->ch[0]);
 62     pool[++tot]=o;
 63     if(o->ch[1]!=null)Erholung(o->ch[1]);
 64 }
 65 inline void travel(Goat *o)
 66 {
 67     if(o==null)return;
 68     if(o->ch[0]!=null)travel(o->ch[0]);
 69     sta[++top]=o;
 70     if(o->ch[1]!=null)travel(o->ch[1]);
 71 }
 72 inline Goat* build(int l,int r)
 73 {
 74     if(l>r)return null;
 75     int mi=l+r>>1;
 76     Goat *o=sta[mi];o->size=r-l+1;
 77     o->ch[0]=build(l,mi-1),o->ch[1]=build(mi+1,r);
 78     return o;
 79 }
 80 inline void rebuild(Goat *&o){top=0,travel(o),o=build(1,top);}
 81 inline void Insert(Goat *&a,int val)
 82 {
 83     Goat **o=insert(a,val);
 84     if(*o!=null)rebuild(*o);
 85 }
 86 int size[N],maxs[N],totsize,root,Vater[N];
 87 #define max(a,b) ((a)>(b)?(a):(b))
 88 #define min(a,b) ((a)<(b)?(a):(b))
 89 bool vis[N];
 90 inline void dfs1(int rt,int fa)
 91 {
 92     size[rt]=1,maxs[rt]=0;
 93     for(int i=adj[rt];i;i=s[i].next)
 94         if(!vis[s[i].zhong]&&s[i].zhong!=fa)
 95             dfs1(s[i].zhong,rt),size[rt]+=size[s[i].zhong],
 96             maxs[rt]=max(maxs[rt],size[s[i].zhong]);
 97     maxs[rt]=max(maxs[rt],totsize-size[rt]);
 98     if(maxs[rt]<maxs[root])root=rt;
 99 }
100 inline void dfs2(int rt,int fa,int Vatti)
101 {
102     Insert(tree1[Vatti],dis(rt,Vatti)-val[rt]);
103     if(Vater[Vatti])Insert(tree2[Vatti],dis(rt,Vater[Vatti])-val[rt]);
104     for(int i=adj[rt];i;i=s[i].next)
105         if(s[i].zhong!=fa&&!vis[s[i].zhong])
106             dfs2(s[i].zhong,rt,Vatti);
107 }
108 inline void dfs3(int rt,int fa)
109 {
110     Vater[rt]=fa,vis[rt]=1;
111     int siz=totsize;
112     dfs2(rt,0,rt);
113     for(int i=adj[rt];i;i=s[i].next)
114         if(!vis[s[i].zhong])
115         {
116             if(size[s[i].zhong]>size[rt])totsize=siz-size[rt];
117             else totsize=size[s[i].zhong];
118             root=0,dfs1(s[i].zhong,rt),
119             to[rt].push_back(root),dfs3(root,rt);
120         }
121 }
122 inline void recover(int x)
123 {
124     ++totsize,vis[x]=0,
125     Erholung(tree1[x]),Erholung(tree2[x]),
126     tree1[x]=tree2[x]=null;
127     for(int i=0,k=to[x].size();i<k;++i)recover(to[x][i]);
128     to[x].clear();
129 }
130 inline void rebuild(int x)
131 {
132     totsize=0,recover(x),root=0,dfs1(x,0);
133     if(Vater[x])for(int i=0,j=to[Vater[x]].size();i<j;++i)
134         if(to[Vater[x]][i]==x)to[Vater[x]][i]=root;
135     dfs3(root,Vater[x]);
136 }
137 LL ans=0;
138 inline int insert(int x)
139 {
140     int rt,ds,ret=0;
141     for(rt=x;Vater[rt];rt=Vater[rt])
142         ds=val[x]-dis(x,Vater[rt])+1,ans+=get_rank(tree1[Vater[rt]],ds)-get_rank(tree2[rt],ds);
143     for(rt=x;rt;rt=Vater[rt])
144     {
145         Insert(tree1[rt],dis(x,rt)-val[x]);
146         if(Vater[rt])Insert(tree2[rt],dis(x,Vater[rt])-val[x]);
147     }
148     for(rt=x;Vater[rt];rt=Vater[rt])
149         if(tree1[rt]->size>=tree1[Vater[rt]]->size*alpha+5)ret=Vater[rt];//*****
150     return ret;
151 }b 
152 char B[1<<15],*S=B,*T=B;
153 #define getc (S==T&&(T=(S=B)+fread(B,1,1<<15,stdin),S==T)?0:*S++)
154 inline int read()
155 {
156     int x=0;register char c=getc;
157     while(c<'0'||c>'9')c=getc;
158     while(c>='0'&&c<='9')x=10*x+(c^48),c=getc;
159     return x;
160 }
161 int main()
162 {
163     n=read(),n=read();
164     register int i,j,x,b;
165     for(bin[0]=i=1;i<=20;++i)bin[i]=bin[i-1]<<1;
166     while(bin[tp+1]<=n)++tp;
167     maxs[0]=inf,root=0,intn();
168     for(i=1;i<=n;++i)
169     {
170         Vater[i]=f[i][0]=read()^(ans%inf),b=read(),val[i]=read();
171         deep[i]=deep[f[i][0]]+1,len[i]=len[f[i][0]]+b;vis[i]=1;
172         if(Vater[i])to[Vater[i]].push_back(i),add(f[i][0],i),add(i,f[i][0]);
173         for(int j=1;bin[j]+1<=deep[i];++j)f[i][j]=f[f[i][j-1]][j-1];
174         x=insert(i);if(x)rebuild(x);
175         printf("%lld\n",ans);
176     }
177 }
BZOJ3435

三.总结

点分治和动态树分治其实是对树规思想的变形。我们利用分治树高度很小的优点,对每个点维护相应的信息,

通过$O(logn)$的额外复杂度来进行普通树规无法完成的操作。希望本文能对你有帮助!

posted @ 2018-01-22 09:24  LadyLex  阅读(3352)  评论(16编辑  收藏  举报