标签：剖分 int siz top pos 树链 id 模板 define
功能:
修改与查询整个子树
修改与查询路径
#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define see(x) (cerr<<(#x)<<'='<<(x)<<endl)
#define pb push_back
#define inf 0x3f3f3f3f
#define CLR(A,v) memset(A,v,sizeof A)
#define lson l,m,pos<<1
#define rson m+1,r,pos<<1|1
typedef pair<int,int>pii;
//////////////////////////////////
const int N=2e6+10;
const int M=1e6;
int t[N<<2],col[N<<2],w[N],n,mod,m,root,node[N];
void up(int pos)
{
t[pos]=t[pos<<1]+t[pos<<1|1];t[pos]%=mod;
}
void down(int m,int pos)
{
if(col[pos])
{
col[pos<<1]+=col[pos];
col[pos<<1|1]+=col[pos];
t[pos<<1]+=(m-(m>>1))*col[pos];t[pos<<1]%=mod;
t[pos<<1|1]+=(m>>1)*col[pos];t[pos<<1|1]%=mod;
col[pos]=0;
}
}
void build(int l,int r,int pos)
{
col[pos]=0;
if(l==r){t[pos]=w[l]%mod;return ;}//w为重新编号的权值
int m=(l+r)>>1;
build(lson);build(rson);
up(pos);
}
void update(int L,int R,int x,int l,int r,int pos)
{
if(L<=l&&r<=R)
{
t[pos]+=x*(r-l+1);t[pos]%=mod;col[pos]+=x;return ;
}
down(r-l+1,pos);
int m=(l+r)>>1;
if(L<=m)update(L,R,x,lson);
if(R>m)update(L,R,x,rson);
up(pos);
}
int query(int L,int R,int l,int r,int pos)
{
int ans=0;
if(L<=l&&r<=R)return t[pos];
down(r-l+1,pos);
int m=(l+r)>>1;
if(L<=m)ans+=query(L,R,lson),ans%=mod;
if(R>m)ans+=query(L,R,rson),ans%=mod;
up(pos);
return ans;
}
///////////////////////////////
int dep[N],siz[N],fa[N],son[N],head[M],pos,id[N],top[N],cnt;
struct Edge
{
int to,nex;
}edge[M];
void add(int a,int b)
{
edge[++pos].nex=head[a];
edge[pos].to=b;
head[a]=pos;
}
void dfs1(int x,int f,int deep)
{
dep[x]=deep;
fa[x]=f;
siz[x]=1;
int maxson=-1;
for(int i=head[x];i;i=edge[i].nex)
{
int v=edge[i].to;
if(v==f)continue;
dfs1(v,x,deep+1);
siz[x]+=siz[v];
if(siz[v]>maxson)maxson=siz[v],son[x]=v;
}
}
void dfs2(int x,int topf)
{
id[x]=++cnt;
w[cnt]=node[x];
top[x]=topf;
if(!son[x])return;
dfs2(son[x],topf);
for(int i=head[x];i;i=edge[i].nex)
{
int v=edge[i].to;
if(v==fa[x]||v==son[x])continue;
dfs2(v,v);
}
}
int qrange(int x,int y)//询问x到y最短路径的权值和
{
int ans=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
ans+=query(id[top[x]],id[x],1,n,1);
ans%=mod;
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
ans=(ans+query(id[x],id[y],1,n,1))%mod;
return ans;
}
void updaterange(int x,int y,int k)//将x 到y的最短路径上加k
{
k%=mod;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
update(id[top[x]],id[x],k,1,n,1);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
update(id[x],id[y],k,1,n,1);
}
int qson(int x)//询问以x为根结点的子树去权值和
{
return query(id[x],id[x]+siz[x]-1,1,n,1);
}
void updateson(int x,int k)//将以x为根结点子树所有点的权值加k
{
update(id[x],id[x]+siz[x]-1,k,1,n,1);
}
int main()
{
cin>>n>>m>>root>>mod;
rep(i,1,n)RI(node[i]);
rep(i,1,n-1){int u,v;RII(u,v);add(u,v);add(v,u);}
dfs1(root,0,1);
dfs2(root,root);
build(1,n,1);
rep(i,1,m)
{
int k,x,y,z;RI(k);
if(k==1)
{
RIII(x,y,z);updaterange(x,y,z);
}
else if(k==2)
{
RII(x,y);printf("%d\n",qrange(x,y));
}
else if(k==3)
{
RII(x,y);updateson(x,y);
}
else
{
RI(x);printf("%d\n",qson(x));
}
}
return 0;
}
View Code
标签：剖分,int,siz,top,pos,树链,id,模板,define 来源： https://www.cnblogs.com/bxd123/p/11162416.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。