刷题记录:牛客NC19428树上求和 树链剖分+区间平方和

传送门:牛客

题目描述:

给你一棵根为1的有N个节点的树，以及Q次操作。
每次操作诸如:
1 x y：将节点x所在的子树的所有节点的权值加上y
2 x：询问x所在子树的所有节点的权值的平方和，答案模23333后输出
输入:
5 5
0 0 0 0 0
1 2
1 3
3 4
3 5
1 1 3
1 3 7
1 4 5
1 5 6
2 1
输出:
599

一道树链剖分+线段树的题目

假如不是树形结构,这道题就和此题解决方法类似了,直接使用线段树进行维护区间修改+区间和即可

但是本题是树形结构,所以按照套路,我们可以使用树链剖分将树形结构分解为线性结构然后就可以直接使用线段树进行维护啦

对于一个节点的所有子树来说,因为树链剖分的特性,一个节点的所有子树在线段树的存储位置必然是连续的,所以我们直接更新$[id[u],id[u]+Size[u]-1]$即可

下面是具体代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define int long long
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],Size[maxn],dep[maxn],max_son[maxn];
vector<int>edge[maxn];
const int mod=23333;
void dfs1(int u,int per_u) {
	Size[u]=1;
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==per_u) continue;
		dep[v]=dep[u]+1;
		dfs1(v,u);
		Size[u]+=Size[v];fa[v]=u;
		if(Size[v]>Size[max_son[u]]) {
			max_son[u]=v;
		}
	}
}
int top[maxn],rev[maxn],id[maxn],tot=0;
void dfs2(int u,int t) {
	top[u]=t;id[u]=++tot;rev[tot]=u;
	if(!max_son[u]) return ;
	dfs2(max_son[u],t);
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==fa[u]||v==max_son[u]) continue;
		dfs2(v,v);
	}
}
struct Segment_tree{
	int l,r,sq_sum,lazy,sum;
}tree[maxn*4];int w[maxn];
void pushup(int rt) {
	tree[rt].sq_sum=tree[ls].sq_sum+tree[rs].sq_sum;
	tree[rt].sum=tree[ls].sum+tree[rs].sum;
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;
	if(l==r) {
		tree[rt].sq_sum=w[rev[l]]*w[rev[l]];
		tree[rt].sum=w[rev[l]];
		return ;
	}
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(rt);
}
void change(int rt,int val) {
	int len=tree[rt].r-tree[rt].l+1;
	tree[rt].sq_sum=(tree[rt].sq_sum+val*val*len%mod+2*tree[rt].sum*val%mod)%mod;
	tree[rt].sum=(tree[rt].sum+val*len%mod)%mod;
	tree[rt].lazy+=val;
}
void pushdown(int rt) {
	change(ls,tree[rt].lazy);change(rs,tree[rt].lazy);
	tree[rt].lazy=0;
}
void update(int l,int r,int rt,int val) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		change(rt,val);
		return ;
	}
	if(tree[rt].lazy) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) update(l,r,ls,val);
	else if(l>mid) update(l,r,rs,val);
	else update(l,mid,ls,val),update(mid+1,r,rs,val);
	pushup(rt);
}
int ans_sum=0;
void query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		ans_sum=(ans_sum+tree[rt].sq_sum)%mod;
		return ;
	}
	if(tree[rt].lazy) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) query(l,r,ls);
	else if(l>mid) query(l,r,rs);
	else query(l,mid,ls),query(mid+1,r,rs);
}
int n,q;
signed main() {
	n=read(),q=read();
	for(int i=1;i<=n;i++) w[i]=read();
	for(int i=1;i<=n-1;i++) {
		int u=read(),v=read();
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	dfs1(1,0);dfs2(1,1);
	build(1,tot,1);
	for(int i=1;i<=q;i++) {
		int opt=read();
		if(opt==1) {
			int u=read(),val=read();
			update(id[u],id[u]+Size[u]-1,1,val);
		}
		else {
			int u=read();
			ans_sum=0;
			query(id[u],id[u]+Size[u]-1,1);
			printf("%lld\n",ans_sum);
		}
	}
	return 0;
}