luogu P4074 [WC2013]糖果公园（树上带修莫队）

最新推荐文章于 2021-08-15 23:11:43 发布

Qingo呀

最新推荐文章于 2021-08-15 23:11:43 发布

阅读量208

点赞数

分类专栏： =====莫队===== =====模板=====

本文链接：https://blog.csdn.net/birdmanqin/article/details/110404353

版权

=====模板===== 同时被 2 个专栏收录

67 篇文章 0 订阅

订阅专栏

=====莫队=====

19 篇文章 0 订阅

订阅专栏

博客主要介绍了如何利用欧拉序和莫队算法解决区间修改查询问题。首先将区间处理成连续形式，然后采用类似于带修莫队的方法进行处理。在修改操作中，要注意判断元素是否已被修改过，根据状态进行增删处理，并考虑LCA的贡献。最后，通过排序和块状结构优化查询效率。博客内容涉及数据结构和算法，特别是图论和区间查询的相关知识。

摘要由CSDN通过智能技术生成

题链：https://www.luogu.com.cn/problem/P4074

思路：用欧拉序处理成连续区间后，就和带修莫队差不多了。

要注意的是，对于修改。要看看是否之前有没有被加过，加过（也就是vis[i]为1）的要减去，然后再加上新的；如果没加过就不用管。最后再加上lca的贡献就行。注意块的大小为 $n^{\frac{2}{3}}$ 。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5+10;
const int M = 1e6+10;
int n,m,q;
int v[N],w[N];
int a[N];
struct node{
	int l,r,id,tim,lca;
}qu[N];
int cntq=0;
struct Node{
	int pos,val;
}b[N];
int cntb=0;
int block,be[N<<1];
vector<int> g[N];
int Id=0,dep[N],fir[N],la[N],dfn[N<<1],f[N][30],limit;
void dfs(int u,int fa){
	for(int i=1;i<=limit;i++)
		f[u][i]=f[f[u][i-1]][i-1];
	fir[u]=++Id;
	dfn[Id]=u;
	int siz=g[u].size();
	for(int i=0;i<siz;i++){
		int v=g[u][i];
		if(v==fa) continue;
		f[v][0]=u;
		dep[v]=dep[u]+1;
		dfs(v,u);
	}
	la[u]=++Id;
	dfn[Id]=u;	
}
int getlca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	int dis=dep[x]-dep[y];
	for(int i=limit;i>=0;i--)
		if(dis&(1<<i))
			x=f[x][i]; 
	if(x==y) return x;
	for(int i=limit;i>=0;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
bool cmp(node a,node b){
	return (be[a.l]^be[b.l]) ? be[a.l]<be[b.l] : ((be[a.r]^be[b.r]) ? be[a.r]<be[b.r] : a.tim<b.tim);	
}
int num[N];
ll sum=0,ans[N];
void add(int x){
	++num[a[x]];
	sum+=1LL*v[a[x]]*w[num[a[x]]];
}
void del(int x){
	sum-=1LL*v[a[x]]*w[num[a[x]]];
	--num[a[x]];
}
int vis[N];
void change(int x){
	vis[x]^=1;
	if(vis[x]) add(x);
	else del(x);
}
void change1(int x){
	if(vis[x]) del(x);
	else add(x);
}
void change2(int x){
	if(vis[x]) add(x);
	else del(x);
}
#define isdigit(x) ((x) >= '0' && (x) <= '9')
int read() {
	int res = 0;
	char c = getchar();
	while(!isdigit(c)) c = getchar();
	while(isdigit(c)) res = (res << 1) + (res << 3) + c - 48, c = getchar();
	return res;
}
int main(void){
	//scanf("%d%d%d",&n,&m,&q);
	n=read(),m=read(),q=read();
	limit=ceil(log2(1.0*n));
	block=ceil(pow(2.0*n,2.0/3.0));
	for(int i=1;i<=m;i++) v[i]=read();//scanf("%lld",&v[i]);
	for(int i=1;i<=n;i++){
		//scanf("%lld",&w[i]);
		w[i]=read();
		be[i]=i/block;
		be[i+n]=(i+n)/block;	
	} 
	for(int i=1;i<n;i++){
		int u,v;
		//scanf("%d%d",&u,&vv);
		u=read(),v=read();
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dep[1]=1;
	dfs(1,0);
	for(int i=1;i<=n;i++) a[i]=read(); //scanf("%d",&a[i]);	
	 
	for(int i=1;i<=q;i++){
		int ty,u,v;
		//scanf("%d%d%d",&ty,&u,&v);
		ty=read(),u=read(),v=read();		
		if(ty){
			++cntq;
			if(fir[u]>fir[v]) swap(u,v);
			int lca=getlca(u,v);
			if(u==lca) qu[cntq].l=fir[u];
			else qu[cntq].l=la[u],qu[cntq].lca=lca;
			qu[cntq].r=fir[v];
			qu[cntq].id=cntq;
			qu[cntq].tim=cntb;
		}else
			b[++cntb]=Node{u,v};
	}
	sort(qu+1,qu+1+cntq,cmp);
	int l=1,r=0,ti=0;
	for(int i=1;i<=cntq;i++){
		int ql=qu[i].l,qr=qu[i].r,qt=qu[i].tim,lca=qu[i].lca;
		while(l<ql) change(dfn[l++]);
		while(l>ql) change(dfn[--l]);
		while(r<qr) change(dfn[++r]);
		while(r>qr) change(dfn[r--]);
		while(ti<qt){
			++ti;
			if(vis[b[ti].pos])
				del(b[ti].pos);		
			swap(a[b[ti].pos],b[ti].val);
			if(vis[b[ti].pos])
				add(b[ti].pos);			
		}
		while(ti>qt){
			if(vis[b[ti].pos])
				del(b[ti].pos);					
			swap(a[b[ti].pos],b[ti].val);
			if(vis[b[ti].pos])
				add(b[ti].pos);						
			ti--;
		}
		if(lca) change(lca);
		ans[qu[i].id]=sum;
		if(lca) change(lca);		
	}
	for(int i=1;i<=cntq;i++)
		printf("%lld\n",ans[i]);
	return 0;	
}