树分块&&树上莫队&&树上带修莫队

(王室联邦 树分块的裸体  xjb跑dfs就行了

树上莫队个人感觉就是dfs序+普通莫队就可以了

经典例题http://www.spoj.com/problems/COT2/

#include <bits/stdc++.h>
#define ll long long
#define N 40005 
#define MAXN 100005
using namespace std;
ll readll(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int readint(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int n,q,size,p[N<<1],cnt1,cnt2,b[N<<1],cnt3,cnt;
int begin[N],last[N];
ll a[N];
vector<ll>vec;
vector<int>edge[N];
int num[N],son[N],dep[N],pre[N];
void dfs(int v,int fa,int deep){
	b[++cnt3]=v;num[v]=1,dep[v]=deep+1;pre[v]=fa;begin[v]=cnt3;
	for(int i=0;i<edge[v].size();i++){
		int u=edge[v][i];
		if(u!=fa){
			dfs(u,v,deep+1);
			num[v]+=num[u];
			if(son[v]==-1||num[u]>num[son[v]]) son[v]=u;
		}
	}
	b[++cnt3]=v;last[v]=cnt3;
}
int tp[N];
void dfs2(int v,int td){
	tp[v]=td;
	if(son[v]!=-1) dfs2(son[v],td);
	for(int i=0;i<edge[v].size();i++){
		int u=edge[v][i];
		if(u!=pre[v]&&u!=son[v]) dfs2(u,u);
	}
}
typedef struct node{
	int l,r,biao,lca,flag;
	friend bool operator <(node aa,node bb){
		if(p[aa.l]==p[bb.l]) return aa.r<bb.r;
		return p[aa.l]<p[bb.l];
	}
}node;
int Lca(int u,int v){
	int uu=tp[u];int vv=tp[v];
	while(uu!=vv){
		if(dep[uu]<dep[vv]){
			swap(uu,vv);swap(u,v);
		}
		u=pre[uu];uu=tp[u];
	}
	if(dep[u]>dep[v]) swap(u,v);
	return u;
}
node d[MAXN];int cont[N],vis[N];
int ans[MAXN],sum;
void check(int R){
		vis[b[R]]^=1;
		if(vis[b[R]]){
			cont[a[b[R]]]++;
			if(cont[a[b[R]]]==1) sum++;
		}
		else{
			cont[a[b[R]]]--;
			if(cont[a[b[R]]]==0) sum--;
		}
}
int main(){
	ios::sync_with_stdio(false);cnt1=0;cnt2=0;cnt3=0;cnt=0;
	n=readint();q=readint();size=(int)sqrt(2*n);
	for(int i=1;i<=n;i++) son[i]=-1;
	for(int i=1;i<=n;i++) a[i]=readll(),vec.push_back(a[i]);
	for(int i=1;i<=2*n;i++) p[i]=(i-1)/size+1;
	sort(vec.begin(),vec.end());
	int t=unique(vec.begin(),vec.end())-vec.begin();
	for(int i=1;i<=n;i++) a[i]=lower_bound(vec.begin(),vec.begin()+t,a[i])-vec.begin()+1;
	int u,v;
	for(int i=1;i<n;i++){
		u=readint();v=readint();
		edge[u].push_back(v);edge[v].push_back(u);
	}
	dfs(1,-1,0);dfs2(1,1);
	int l,r;
	for(int i=1;i<=q;i++){
		l=readint();r=readint();d[i].lca=Lca(l,r);d[i].biao=i;
		if(d[i].lca==l||d[i].lca==r) d[i].l=min(begin[l],begin[r]),d[i].r=max(begin[l],begin[r]),d[i].flag=1;
		else{
			if(begin[l]<begin[r]) d[i].l=last[l],d[i].r=begin[r];
			else d[i].l=last[r],d[i].r=begin[l];
		}
	}
	sort(d+1,d+q+1);
	int L=1,R=0;sum=0;
	for(int i=1;i<=q;i++){
		while(R>d[i].r){
			check(R);R--;
		}
		while(R<d[i].r){
			R++;check(R);
		}
		while(L>d[i].l){
			L--;check(L);
		}
		while(L<d[i].l){
			check(L);L++;
		}
		if(d[i].flag) ans[d[i].biao]=sum;
		else{
			ans[d[i].biao]=sum;
			if(!cont[a[d[i].lca]]) ans[d[i].biao]++;
		}
	}
	for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
	return 0;
}

　(对于树上带修莫队  会了树上莫队  加上一个带修操作 和普通带修莫队没啥区别 复杂度其他证明也基本类似

经典例题http://www.lydsy.com/JudgeOnline/problem.php?id=3052

#include <bits/stdc++.h>
#define ll long long
#define N 100005
#define MAXN 1000005
#define pii pair<int,int> 
using namespace std;
ll readll(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int readint(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int n,m,q;
int a[N],p[N<<1],c[N];
ll V[N],W[N];
int b[N<<1],begin[N],last[N],cnt,fa[N],dep[N],size;
int num[N],son[N],first[N],cnt1;
pii Next[N<<1];
void add(int u,int v){
	Next[++cnt1].first=first[u];Next[cnt1].second=v;first[u]=cnt1;
	Next[++cnt1].first=first[v];Next[cnt1].second=u;first[v]=cnt1;
}
void dfs(int u,int pre,int deep){
	num[u]=1;dep[u]=deep+1;fa[u]=pre;b[++cnt]=u;begin[u]=cnt;
	for(int i=first[u];i!=-1;i=Next[i].first){
		int v=Next[i].second;
		if(v!=pre){
			dfs(v,u,deep+1);
			num[u]+=num[v];
			if(son[u]==-1||num[v]>num[son[u]]) son[u]=v;
		}
	}
	b[++cnt]=u;last[u]=cnt;
}
int tp[N];
void dfs1(int v,int td){
	tp[v]=td;
	if(son[v]!=-1) dfs1(son[v],td);
	for(int i=first[v];i!=-1;i=Next[i].first){
		int u=Next[i].second;
		if(u!=fa[v]&&u!=son[v]) dfs1(u,u);
	} 
}
int Lca(int u,int v){
	int uu=tp[u];int vv=tp[v];
	while(uu!=vv){
		if(dep[uu]<dep[vv]){
			swap(uu,vv);swap(u,v);
		}
		u=fa[uu];uu=tp[u];
	}
	if(dep[u]>dep[v]) swap(u,v);
	return u;
} 
typedef struct node{
	int l,r,lca,biao;int t;bool flag;
	friend bool operator <(node aa,node bb){
		if(p[aa.l]==p[bb.l]&&p[aa.r]==p[bb.r]) return aa.t<bb.t;
		else if(p[aa.l]==p[bb.l]) return p[aa.r]<p[bb.r];
		else return p[aa.l]<p[bb.l];
	}
}node;
node d[N];int vis[N],cont[1000006];
pii to[N],past[N];
ll ans[N],sum;
void work(int pos){
	vis[b[pos]]^=1;
	if(vis[b[pos]]){
		cont[a[b[pos]]]++;sum+=1ll*W[cont[a[b[pos]]]]*V[a[b[pos]]];
	}
	else{
		cont[a[b[pos]]]--;sum-=1ll*W[cont[a[b[pos]]]+1]*V[a[b[pos]]];
	}
}
void update(int l,int r,int t){
	int u=to[t].first;int vul=to[t].second;
//	cout<<u<<"----"<<vul<<endl;
	int t1=begin[u];int t2=last[u];
	int cp=0;int pos;
	if(t1>=l&&t1<=r) cp++,pos=t1;
	if(t2>=l&&t2<=r) cp++,pos=t2;
//	cout<<a[b[pos]]<<" "<<cp<<endl;
	if(cp==1){
		cont[a[b[pos]]]--;;sum-=1ll*W[cont[a[b[pos]]]+1]*V[a[b[pos]]];
		//cout<<sum<<endl;
		cont[vul]++;sum+=1ll*W[cont[vul]]*V[vul];
	}
	a[u]=vul;
}
void restore(int l,int r,int t){
	int u=past[t].first;int vul=past[t].second;
	int t1=begin[u];int t2=last[u];
	int cp=0;int pos;
	if(t1>=l&&t1<=r) cp++,pos=t1;
	if(t2>=l&&t2<=r) cp++,pos=t2;
	if(cp==1){
		cont[a[b[pos]]]--;;sum-=1ll*W[cont[a[b[pos]]]+1]*V[a[b[pos]]];
		cont[vul]++;sum+=1ll*W[cont[vul]]*V[vul];
	}
	a[u]=vul;
}
int main(){
	ios::sync_with_stdio(false);cnt=cnt1=0;
	//cout<<size<<endl;
	n=readint();m=readint();q=readint();
	size=(int)pow(2*n,2.0/3);	
	for(int i=1;i<=2*n;i++) p[i]=(i-1)/size+1;;
	for(int i=1;i<=n;i++) son[i]=first[i]=-1;
	for(int i=1;i<=m;i++) V[i]=readll();
	for(int i=1;i<=n;i++) W[i]=readll();
	int u,v;
	for(int i=1;i<n;i++){
		u=readint();v=readint();
		add(u,v);
	}
	for(int i=1;i<=n;i++) c[i]=a[i]=readint();
	dfs(1,-1,0);dfs1(1,1);
	//for(int i=1;i<=2*n;i++) cout<<b[i]<<" ";
	//cout<<endl;
	int op,l,r;int T=0;u=0;
	for(int i=1;i<=q;i++){
		op=readint();l=readint();r=readint();
		if(op==0){
			to[++T].first=l;to[T].second=r;
			past[T].first=l;past[T].second=c[l];
			c[l]=r;
		}
		else{
			d[++u].lca=Lca(l,r);d[u].biao=u;d[u].t=T;
			if(d[u].lca==l||d[u].lca==r) d[u].l=min(begin[l],begin[r]),d[u].r=max(begin[l],begin[r]);
			else{
				if(begin[l]>begin[r]) swap(l,r);
				d[u].l=last[l];d[u].r=begin[r];d[u].flag=1;
			}
		}
	}
//	cout<<u<<endl;
	sort(d+1,d+u+1);int L=1,R=0,t=0;sum=0;
//	for(int i=1;i<=u;i++){
//		cout<<d[i].l<<" "<<d[i].r<<" "<<d[i].lca<<" "<<d[i].t<<endl;
//	}
	//for(int i=1;i<=2*n;i++) cout<<b[i]<<" ";
//	cout<<endl;
	for(int i=1;i<=u;i++){
	//	cout<<d[i].l<<"====="<<d[i].r<<" "<<d[i].t<<"======"<<d[i].biao<<endl;
		while(t>d[i].t){
			restore(L,R,t);t--;
		}
		while(t<d[i].t){
			t++;update(L,R,t);
		}
		while(R>d[i].r){
			work(R);R--;
		}
		while(R<d[i].r){
			R++;work(R);
		}
		while(L<d[i].l){
			work(L);L++;
		}
		while(L>d[i].l){
			L--;work(L);
		}
		ans[d[i].biao]=sum;
		if(d[i].flag) ans[d[i].biao]+=(W[cont[a[d[i].lca]]+1]*V[a[d[i].lca]]);
	}
	for(int i=1;i<=u;i++) printf("%lld\n",ans[i]);
	return 0;
}