标签：leq int top long son 牛客 57C 挑战赛 500005
牛客挑战赛57C树上行走
题目大意
给一棵\({n}\)个点的树，每个点都有一个初始的权值\({a_i}\)，有一个计数器\({b_i}\)，初始值为\({0}\)，现有两种操作：\({1、}\)给定\({x,y}\)，令\({x\rarr y}\)的最短路上的点构成的序列为\({p}\)，\({\forall i>1}\)，\({b_{p_i}}\)增加\({a_{p_{i-1}}}\)；\({2、}\)给定\({x}\)，输出\({b_x}\)的值。\({(1\leq n\leq 5e5,1\leq q\leq 3e5,0\leq a_i\leq 1e9)}\)
思路
很容易想到树链剖分，但是新手如我，之后就不会了。不过幸好，\({canon}\)大佬给了一点思路。首先我们当然不可能暴力的去更新每一个点的值，那么我们可以发现能对一个点产生贡献的无非是它的父亲、重儿子和轻儿子，然后我们发现每个点的父亲和重儿子都最多只有一个，轻儿子可以有很多但是一棵树种轻儿子的数量一定是很少的，所以我们就可以通过暴力更新轻儿子的贡献，而对于父亲和重儿子的贡献我们可以通过建两个数组的方法来进行标记，询问的时候一起加上就可以了。而快速标记的方法无非线段树和树状数组，这两种方法都可以。
树状数组代码
#include<bits/stdc++.h>
using namespace std;
int n,q;
vector<vector<int>>v(500005);
long long a[500005];
int dep[500005],siz[500005],son[500005],fa[500005];
int id[500005],top[500005],tot=0;
long long ans[500005];
long long tree[2][500005];
void dfs(int x,int father)
{
siz[x]=1;
dep[x]=dep[father]+1;
int maxsize=0;
for(auto it:v[x])if(it!=father)
{
fa[it]=x;
dfs(it,x);
siz[x]+=siz[it];
if(siz[it]>maxsize)
{
maxsize=siz[it];
son[x]=it;
}
}
}
void dfs1(int x,int father)
{
id[x]=++tot;
if(son[x])
{
top[son[x]]=top[x];
dfs1(son[x],x);
}
for(auto it:v[x])if(it!=father&&it!=son[x])
{
top[it]=it;
dfs1(it,x);
}
}
int lca(int x,int y)
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
x=fa[top[x]];
}
if(dep[x]<dep[y])return x;
else return y;
}
void update(int x,int w,int op)
{
while(x<=500005)
{
tree[op][x]+=w;
x+=(x&-x);
}
}
int query(int x,int op)
{
int sum=0;
while(x)
{
sum+=tree[op][x];
x-=(x&-x);
}
return sum;
}
void Update1(int x,int y)
{
while(top[x]!=top[y])
{
if(top[x]!=x)
{
update(id[top[x]],1,0);
update(id[fa[x]]+1,-1,0);
}
x=top[x];
ans[fa[x]]+=a[x];
x=fa[x];
}
if(x!=y)
{
update(id[y],1,0);
update(id[fa[x]]+1,-1,0);
}
}
void Update2(int x,int y)
{
while(top[x]!=top[y])
{
update(id[top[y]],1,1);
update(id[y]+1,-1,1);
y=fa[top[y]];
}
update(id[x]+1,1,1);
update(id[y]+1,-1,1);
}
int main()
{
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
for(int i=1;i<n;i++)
{
int from,to;
scanf("%d%d",&from,&to);
v[from].push_back(to);
v[to].push_back(from);
}
dfs(1,0);
top[1]=1;
dfs1(1,0);
while(q--)
{
int op;
scanf("%d",&op);
if(op==1)
{
int x,y;
scanf("%d%d",&x,&y);
int Lca=lca(x,y);
if(x!=Lca)Update1(x,Lca);
if(y!=Lca)Update2(Lca,y);
}
else
{
int x;
scanf("%d",&x);
long long ans1=ans[x];
if(son[x])ans1+=a[son[x]]*query(id[x],0);
if(fa[x])ans1+=a[fa[x]]*query(id[x],1);
printf("%lld\n",ans1);
}
}
return 0;
}
线段树代码
//现在搁这儿晾会儿，等我改好会更的哦~
标签：leq,int,top,long,son,牛客,57C,挑战赛,500005 来源： https://www.cnblogs.com/Jerry-Black/p/15921374.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。