洛谷 P3899 [谈笑风生]

简化题意

m次询问，每次询问x的子树中，与x节点距离不超过y的节点的子树和。n,m≤300,000。

---

 

思路

按照dfs序排序，每次将一个点的答案塞到第dep_u的位置，这样得到一个前缀和，每次询问作减法即可。

可持久化线段树。

---

 

代码

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn=3E5+5;
 4 typedef long long int ll;
 5 ll n,m,x,y,sum[maxn],dep[maxn],cur,t[maxn*20],root[maxn*20],fa[maxn],size,gg;
 6 ll ls[maxn*20],rs[maxn*20],ti,dfn[maxn],last[maxn],maxx[maxn],head[maxn*2];
 7 struct edge{ll to,next;}E[maxn*2];
 8 void insert(ll l,ll r,ll pos,ll&num,ll pre,ll val)
 9 {
10     num=++cur;
11     t[num]=t[pre];
12     ls[num]=ls[pre];
13     rs[num]=rs[pre];
14     t[num]+=val;
15     if(l==r)return;
16     ll mid=(l+r)>>1;
17     if(pos<=mid)insert(l,mid,pos,ls[num],ls[pre],val);
18     else insert(mid+1,r,pos,rs[num],rs[pre],val);
19 }
20 ll ask(ll L,ll R,ll l,ll r,ll num)
21 {
22     if(L<=l&&r<=R)return t[num];
23     else if(r<L||R<l)return 0;
24     ll mid=(l+r)>>1;
25     return ask(L,R,l,mid,ls[num])+ask(L,R,mid+1,r,rs[num]);
26 }
27 void add(ll u,ll v)
28 {
29     E[++size].to=v;
30     E[size].next=head[u];
31     head[u]=size;
32 }
33 void dfs(ll u,ll F,ll d)
34 {
35     dep[u]=d;
36     fa[u]=F;
37     sum[u]=1;
38     maxx[u]=dfn[u]=++ti;
39     last[u]=u;
40     for(ll i=head[u];i;i=E[i].next)
41     {
42         ll v=E[i].to;
43         if(v==F)continue;
44         dfs(v,u,d+1);
45         sum[u]+=sum[v];
46         if(maxx[v]>maxx[u])
47         {
48             maxx[u]=maxx[v];
49             last[u]=last[v];
50         }
51     }
52 }
53 void init(ll u,ll F,ll d)
54 {
55     insert(1,n,d,root[u],root[gg],sum[u]-1);
56     gg=u;
57     for(ll i=head[u];i;i=E[i].next)
58     {
59         ll v=E[i].to;
60         if(v==F)continue;
61         init(v,u,d+1);
62     }
63 }
64 int main()
65 {
66 //    freopen("laugh.in","r",stdin);
67 //    freopen("laugh.out","w",stdout);
68     ios::sync_with_stdio(false);
69     cin>>n>>m;
70     for(ll i=1;i<=n-1;++i)
71     {
72         cin>>x>>y;
73         add(x,y);
74         add(y,x);
75     }
76     dfs(1,1,1);
77     gg=1;
78     init(1,1,1);
79     for(ll i=1;i<=m;++i)
80     {
81         cin>>x>>y;
82         if(x==1)cout<<ask(2,1+y,1,n,root[last[1]])<<endl;
83         else cout<<(sum[x]-1)*min(dep[x]-1,y)+
84         ask(dep[x]+1,dep[x]+y,1,n,root[last[x]])-ask(dep[x]+1,dep[x]+y,1,n,root[x])<<endl;
85     }
86     return 0;
87 }

View Code

 

转载于:https://www.cnblogs.com/GreenDuck/p/10610636.html