【BZOJ4372】烁烁的游戏(动态点分治)(树状数组)

传送门


题解:

建立点分树,每个分治中心开一个树状数组,表示它对所有到它距离为tt的点的贡献

每次修改直接暴跳分治树父亲修改,发现父子分治中心会对一些点有重复贡献,每个儿子开一个树状表示它的连通块内距离父亲不超过tt的点被重复贡献了多少。

询问暴跳点分树查每一个分治中心的树状数组即可。


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

namespace IO{
	static cs int Rlen=1<<22|1;
	static char buf[Rlen],*p1,*p2;
	inline char gc(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;}
	inline char peek(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1;}
	inline char ga(){while(!isalpha(peek()))++p1;return *p1++;}
	template<typename T>
	inline T get(){
		char c;T num;bool f=0;
		while(!isdigit(c=gc()))f=c=='-';num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return f?-num:num;
	}
	inline int gi(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int N=1e5+7;

int n,m;

std::vector<int> G[N];
inline void adde(int u,int v){
	G[u].push_back(v);G[v].push_back(u);
} 

namespace ST{
	int in[N],dfn,d[N],mn[19][N<<1],Log[N<<1];
	void dfs(int u,int p){
		mn[0][in[u]=++dfn]=d[u]=d[p]+1;
		for(int re v:G[u])if(v!=p)dfs(v,u),mn[0][++dfn]=d[u];
	}
	inline void init(){
		dfs(1,0);for(int re i=2;i<=dfn;++i)Log[i]=Log[i>>1]+1;
		for(int re i=1;i<=Log[dfn];++i)
		for(int re j=1;j+(1<<i)-1<=dfn;++j)
		mn[i][j]=std::min(mn[i-1][j],mn[i-1][j+(1<<i-1)]); 
	}
	inline int dis(int u,int v){
		int l=in[u],r=in[v];if(l>r)std::swap(l,r);
		int t=Log[r-l+1];return d[u]+d[v]-2*std::min(mn[t][l],mn[t][r-(1<<t)+1]);
	}
}
using ST::dis;

int fa[N],ban[N],siz[N];
int rt,total,mx_siz;

void get_siz(int u,int p){
	siz[u]=1;for(int re v:G[u])
	if(v!=p&&!ban[v]){get_siz(v,u);siz[u]+=siz[v];}
}
void find_rt(int u,int p){
	int mx=total-siz[u];for(int re v:G[u])
	if(v!=p&&!ban[v]){find_rt(v,u);mx=std::max(mx,siz[v]);}
	if(mx<mx_siz){mx_siz=mx,rt=u;}
}
void get_rt(int u){
	get_siz(u,0);
	total=siz[u],mx_siz=0x3f3f3f3f;
	find_rt(u,0);
}

struct BIT{
	std::vector<int> a;int n;
	inline void init(int _n){n=_n+1;a.resize(n+1);}
	inline void add(int p,int v){for(p=std::min(p+1,n);p>=1;p^=p&-p)a[p]+=v;}
	inline int query(int p){int res=0;for(p=std::min(p+1,n);p<=n;p+=p&-p)res+=a[p];return res;}
}f1[N],f2[N];

int mxd;
void get_d(int u,int p,int d){
	mxd=std::max(d,mxd);
	for(int re v:G[u])if(v!=p&&!ban[v])get_d(v,u,d+1);
}

void DFS(int u){
	ban[u]=true;mxd=0;
	get_d(u,0,0);f1[u].init(mxd);
	for(int re v:G[u])if(!ban[v]){
		get_rt(v);fa[rt]=u;
		mxd=0;get_d(v,0,1);
		f2[rt].init(mxd);
		DFS(rt);
	}
}

inline int query(int u){
	int ans=0;
	for(int re v=u;v;v=fa[v]){
		ans+=f1[v].query(dis(u,v));
		if(fa[v])ans-=f2[v].query(dis(u,fa[v]));
	}
	return ans;
}

inline void modify(int u,int d,int w){
	for(int re v=u;v;v=fa[v]){
		f1[v].add(d-dis(u,v),w);
		if(fa[v])f2[v].add(d-dis(u,fa[v]),w);
	}
}

signed main(){
#ifdef zxyoi
	freopen("game.in","r",stdin);
#endif
	n=gi(),m=gi();
	for(int re i=1;i<n;++i)adde(gi(),gi());
	ST::init();get_rt(1);DFS(rt);
	while(m--){
		switch(ga()){
			case 'Q':cout<<query(gi())<<"\n";break;
			case 'M':{
				int u=gi(),d=gi(),w=gi();
				modify(u,d,w);break;
			}
		}
	}
	return 0;
}
发布了1034 篇原创文章 · 获赞 378 · 访问量 9万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览