bzoj 3779 重组病毒 好题 LCT+dfn序+线段树分类讨论

题目大意

1、将x到当前根路径上的所有点染成一种新的颜色;
2、将x到当前根路径上的所有点染成一种新的颜色,并且把这个点设为新的根;
3、查询以x为根的子树中所有点权值的平均值。

分析

原题codechef ,Gangsters of Treeland
那题没有换根操作
用神转化把问题转操作1转化成access操作
操作3转化成每个点到根上有多少条虚边
用dfn序+线段树维护
现在多了个换根操作,只是线段树上加个分类讨论而已

注意

longdouble会Wa,double就A了

姿势

1.用dfn序判断x是否y的祖先

bool ispre(int x,int y){
	if(x==y) return 0;//相等时不是祖先
	return bg[x]<=bg[y]&&ed[y]<=ed[x];
}

2.分类讨论姿势

xxx(int x){
	if(x==rt) xxx;
	else if(ispre(x,rt)){
		xxx;
	}
	else{
		xxx;
	}
}

3.多种数据结构时,用类似 seg_mdf()这样加前缀和下划线的方法
4.以后splay打翻转标记时就把儿子换了吧
5.可用注释分割代码

solution

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <algorithm>
using namespace std;
typedef double db;
typedef long long LL;
const int M=100007;

inline int rd(){
	int x=0;bool f=1;char c=getchar();
	for(;!isdigit(c);c=getchar()) if(c=='-')f=0;
	for(;isdigit(c);c=getchar()) x=x*10+c-48;
	return f?x:-x;
}

int n,m,rt;
int g[M],te;

struct edge{
	int y,nxt;
}e[M<<1];

void addedge(int x,int y){
	e[++te].y=y;e[te].nxt=g[x];g[x]=te;
}

//////////////////////////////// split tree

int top[M],pre[M];
int bg[M],ed[M],tdfn;
int dep[M],sz[M];
int son[M],pid[M];

bool ispre(int x,int y){
	if(x==y) return 0;//*************
	return bg[x]<=bg[y]&&ed[y]<=ed[x];
}

void dfs1(int x){
	sz[x]=1;
	int p,y;
	for(p=g[x];p;p=e[p].nxt)
	if((y=e[p].y)!=pre[x]){
		pre[y]=x;
		dep[y]=dep[x]+1;
		dfs1(y);
		sz[x]+=sz[y];
		if(sz[y]>sz[son[x]]) son[x]=y;
	}
}

void dfs2(int x){
	bg[x]=++tdfn;
	pid[tdfn]=x;
	if(son[x]){
		top[son[x]]=top[x];
		dfs2(son[x]);
	}
	int p,y;
	for(p=g[x];p;p=e[p].nxt)
	if((y=e[p].y)!=pre[x]&&y!=son[x]){
		top[y]=y;
		dfs2(y);
	}
	ed[x]=tdfn;
}

int jump(int x,int to){
	for(;dep[top[x]]>dep[to];x=pre[x]){
		x=top[x];
		if(pre[x]==to) return x;
	}
	return pid[bg[to]+1];	
}

LL getsz(int x){
	if(x==rt) return n;
	if(ispre(x,rt)){
		int y=jump(rt,x);
		return n-sz[y];
	}
	else{
		return sz[x];
	}
}

////////////////////////////////// Segment

struct Seg{
	LL sum,tag;
}c[M<<2];

void seg_pushup(int x){
	c[x].sum=c[x<<1].sum+c[x<<1|1].sum;
}

void seg_totag(int x,LL d,LL len){
	c[x].sum+=d*len;
	c[x].tag+=d;
}

void seg_pushdown(int x,LL aa,LL bb){
	if(c[x].tag){
		seg_totag(x<<1,c[x].tag,aa);
		seg_totag(x<<1|1,c[x].tag,bb);
		c[x].tag=0;
	}
}

void seg_mdf(int x,int l,int r,int tl,int tr,LL d){
	if(tl<=l&&r<=tr){
		seg_totag(x,d,r-l+1);
		return;
	}
	int mid=l+r>>1;
	seg_pushdown(x,mid-l+1,r-mid);
	if(tl<=mid) seg_mdf(x<<1,l,mid,tl,tr,d);
	if(mid<tr) seg_mdf(x<<1|1,mid+1,r,tl,tr,d);
	seg_pushup(x);
}

void seg_add(int x,LL d){
	if(x==rt) return seg_totag(1,d,tdfn);
	if(ispre(x,rt)){
		int y=jump(rt,x);
		seg_mdf(1,1,tdfn,bg[1],ed[1],d);
		seg_mdf(1,1,tdfn,bg[y],ed[y],-d);
	}
	else{
		seg_mdf(1,1,tdfn,bg[x],ed[x],d);
	}
}

LL seg_get(int x,int l,int r,int tl,int tr){
	if(tl<=l&&r<=tr) return c[x].sum;
	int mid=l+r>>1;
	seg_pushdown(x,mid-l+1,r-mid);
	LL res=0;
	if(tl<=mid) res+=seg_get(x<<1,l,mid,tl,tr);
	if(mid<tr) res+=seg_get(x<<1|1,mid+1,r,tl,tr);
	return res;
}

LL seg_sum(int x){
	if(x==rt) return c[1].sum;
	LL res=0;
	if(ispre(x,rt)){
		int y=jump(rt,x);
		res+=seg_get(1,1,tdfn,bg[1],ed[1]);
		res-=seg_get(1,1,tdfn,bg[y],ed[y]);
	}
	else{
		res=seg_get(1,1,tdfn,bg[x],ed[x]);
	}
	return res;
}

///////////////////////////////////////  LCT

struct LCT{
	int ch[2],p,rev;
	int id,lf,rt;//**********************************
	void init(int ii){
		ch[0]=ch[1]=p=rev=0;
		lf=rt=id=ii;
	}
}a[M];

int stack[M],tot;

void torev(int x){
	a[x].rev^=1;
	swap(a[x].ch[0],a[x].ch[1]);//****
	swap(a[x].lf,a[x].rt);
}

void pushup(int x){
	a[x].lf=a[x].rt=a[x].id;
	if(a[x].ch[0]) a[x].lf=a[a[x].ch[0]].lf;
	if(a[x].ch[1]) a[x].rt=a[a[x].ch[1]].rt;
}

void pushdown(int x){
	if(a[x].rev){
		if(a[x].ch[0]) torev(a[x].ch[0]);
		if(a[x].ch[1]) torev(a[x].ch[1]);
		a[x].rev^=1;
	}
}

bool isrt(int x){
	int y=a[x].p;
	return a[y].ch[0]!=x&&a[y].ch[1]!=x;
}

void clear(int x){
	for(;!isrt(x);x=a[x].p) stack[++tot]=x;
	for(stack[++tot]=x;tot>0;tot--) pushdown(stack[tot]);
}

void rot(int x){
	int y=a[x].p;
	int z=a[y].p;
	int D=a[y].ch[1]==x,ss=D^1;
	if(!isrt(y)) a[z].ch[a[z].ch[1]==y]=x;
	a[x].p=z;
	a[y].p=x;
	if(a[x].ch[ss]) a[a[x].ch[ss]].p=y;
	a[y].ch[D]=a[x].ch[ss];
	a[x].ch[ss]=y;
	pushup(y);
	pushup(x);
}

void splay(int x){
	int y,z;
	for(clear(x);!isrt(x);rot(x)){
		y=a[x].p;
		z=a[y].p;
		if(isrt(y)) continue;
		if((a[y].ch[1]==x)!=(a[z].ch[1]==y)) rot(x);
		else rot(y);
	}
}

void access(int x){
	for(int t=0;x;t=x,x=a[x].p){
		splay(x);
		if(a[x].ch[1]) seg_add(a[a[x].ch[1]].lf,1);
		if(t) seg_add(a[t].lf,-1);
		a[x].ch[1]=t;
		pushup(x);
	}
}

void ac(int x){
	access(x);
	splay(x);
}

void chgrt(int x){
	ac(x);
	torev(x);
}

///////////////////////////// main

int main(){
	int i,x,y;
	
	n=rd(),m=rd();rt=1;
	for(i=1;i<n;i++){
		x=rd(),y=rd();
		addedge(x,y);
		addedge(y,x);
	}
	
	pre[1]=0;
	dep[1]=1;
	dfs1(1);
	top[1]=1;
	dfs2(1);
	
	for(i=1;i<=n;i++){
		a[i].init(i);
		if(pre[i]){
			a[i].p=pre[i];
			seg_add(i,1);
		}
	}
	
	char s[13];
	
	while(m--){
		scanf("%s",s);
		x=rd();
		if(s[2]=='L'){
			ac(x);
		}
		else if(s[2]=='C'){
			chgrt(x);
			rt=x;
		}
		else{
			LL tp1=seg_sum(x);
			LL tp2=getsz(x);
			tp1+=tp2;
			
			printf("%.10lf\n",(db)tp1/(db)tp2);
		}
	}
	
	return 0;
}
posted @ 2017-02-21 22:10  _zwl  阅读(711)  评论(0编辑  收藏  举报