[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树

题意题解：
首先对于树上某个点a来说，假设点b是a的祖先（也就是在a的上面），那么答案很好计算，也就是min(k,dep[a]−1)∗(size[a]−1)min(k,dep[a]-1)*(size[a]-1)min(k,dep[a]−1)∗(size[a]−1)

麻烦的就是计算b在a下面的情况。
主席树。
如何建立这颗主席树呢？
按照每个结点的深度建立主席树即可。
对于下面的结点来说，我们假设一个结点iii，那么这个结点对答案的贡献则为size[i]−1size[i]-1size[i]−1
那么我们在dfs序的时候，对于结点iii，插入权值为size[i]−1size[i]-1size[i]−1即可（size为子树大小）
这样就把他给变成了二维数点的问题了。
也就是查询a这颗子树，深度为dep[a]+1到dep[a]+kdep[a]+1到dep[a]+kdep[a]+1到dep[a]+k的和即可。

代码：

#include<bits/stdc++.h>
#define endl '\n'
//#define int long long
using namespace std;const int maxn=3e5+10;int rt[maxn],ls[maxn*50],rs[maxn*50];
long long sum[maxn*50];
int tot,n,q;
void update(int &node,int start,int ends,int lst,int pos,int val){node=++tot;ls[node]=ls[lst];rs[node]=rs[lst];sum[node]=sum[lst]+val;if(start==ends) return;int mid=(start+ends)/2;if(pos<=mid) update(ls[node],start,mid,ls[lst],pos,val);else update(rs[node],mid+1,ends,rs[lst],pos,val);
}
long long query(int start,int ends,int l,int r,int x,int y){if(l<=start&&ends<=r){return sum[y]-sum[x];}int mid=(start+ends)/2;long long ans=0;if(l<=mid) ans+=query(start,mid,l,r,ls[x],ls[y]);if(mid<r) ans+=query(mid+1,ends,l,r,rs[x],rs[y]);return ans;
}vector<int> edge[maxn];
int siz[maxn],son[maxn],in[maxn],fa[maxn],dep[maxn];
int cnt;
void dfs1(int u,int f){dep[u]=dep[f]+1;siz[u]=1;for(auto i:edge[u]){if(i==f) continue;dfs1(i,u);siz[u]+=siz[i];}
}
void build(int u,int f){in[u]=++cnt;update(rt[cnt],1,n,rt[cnt-1],dep[u],siz[u]-1);for(auto i:edge[u]){if(i==f) continue;build(i,u);}
}signed main(){ios::sync_with_stdio(false);cin.tie(0);cin>>n>>q;for(int i=1;i<n;i++){int x,y;cin>>x>>y;edge[x].push_back(y);edge[y].push_back(x);}dfs1(1,0);
//    dfs2(1,1);build(1,0);while(q--){int x,y;cin>>x>>y;long long ans=min(y,dep[x]-1)*1ll*(siz[x]-1);//cout<<"debug "<<ans<<endl;ans+=query(1,n,dep[x]+1,min(dep[x]+y,n),rt[in[x]-1],rt[in[x]+siz[x]-1]);cout<<ans<<endl;//cout<<"debug "<<maxdep<<" "<<dep[x]+1<<" "<<dep[x]+y<<" "<<in[x]-1<<" "<<in[x]+siz[x]-1<<endl;}}

[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树相关推荐

P3899 [湖南集训]更为厉害（线段树合并、长链剖分、二维数点）
P3899 [湖南集训]更为厉害若 deepb<deepa\text{deep}_b<\text{deep}_adeepb<deepa:c 在点 a 的子树中,根据乘法原理计算 ...
【编程题目】输入一颗二元树，从上往下按层打印树的每个结点，同一层中按照从左往右的顺序打印...
第 16 题(树): 题目(微软): 输入一颗二元树,从上往下按层打印树的每个结点,同一层中按照从左往右的顺序打印. 例如输入 8 / \ 6 10 / \ / \ 5 7 9 11 输出 8 6 1 ...
输入一颗二元树，从上向下按层打印树的每个节点，同一层中，按照从左到右的顺序打印。...
输入一颗二元树,从上往下按层打印树的每个结点,同一层中按照从左往右的顺序打印,具体实现如下: #include <iostream> using namespace std;typedef ...
洛谷 - P3899 [湖南集训]谈笑风生(dfs序+主席树/二维数点)
题目链接:点击查看题目大意:设 TTT 为一棵有根树,我们做如下的定义: 设 aaa 和 bbb 为 TTT 中的两个不同节点.如果 aaa 是 bbb 的祖先,那么称"aaa 比 bbb ...
P3899 [湖南集训]谈笑风生（线段树合并）
P3899 [湖南集训]谈笑风生给定一颗以111号节点为根的树,如果a≠ba \neq ba=b,且aaa是bbb的祖先,则aaa比bbb更厉害,如果a≠ba \neq ba=b,且dis( ...
P3899 [湖南集训]谈笑风生
P3899 [湖南集训]谈笑风生题目描述 Solution 我们考虑离线询问,将询问放在相对应的子树ppp中计算答案. 显然a,b,ca,b,ca,b,c的位置关系有两种情况: bbb是aaa的祖先 ...
Super Mario HDU - 4417（主席树解决区间数字小于k的个数||线段树+离线）
Mario is world-famous plumber. His "burly" figure and amazing jumping ability reminded in ...
2020牛客NOIP赛前集训营-提高组(第三场)C-牛半仙的妹子Tree【虚树,最短路】
正题题目链接:https://ac.nowcoder.com/acm/contest/7609/C 题目大意给出nnn个点的一棵树,mmm个时刻各有一个操作标记一个点,每个点被标记后的每一个时刻 ...
Zju2112 Dynamic Rankings（树状数组套可持久化权值线段树）
Zju2112 Dynamic Rankings description solution code description 给定一个含有n个数的序列a[1],a[2],a[3]--a[n],程序必须 ...

[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树

[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树相关推荐

最新文章

热门文章

[湖南集训]更为厉害 树上主席树-以树深度为下下标建立主席树

[湖南集训]更为厉害 树上主席树-以树深度为下下标建立主席树相关推荐

最新文章

热门文章

[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树

[湖南集训]更为厉害树上主席树-以树深度为下下标建立主席树相关推荐