BZOJ 3460 Jc的宿舍

题目链接：http://www.lydsy.com:808/JudgeOnline/problem.php?id=3460

题意：一棵树。每个节点住一个人，这个人打水的时间为Ti。每次查询一个路径。这个路径上的人都去一个水管前打水。总的最小等待时间。

思路：在树的DFS序中分块。

const int N=200005;



struct node

{

	int x,y,lca,flag,id;

	int t;

};



vector<node> b;

vector<int> g[N];

vector<pair<int,int> > Q[N];

int n,m,key,d[N];





int S[N],id;

int mp[N],pos[N][2];





int get(int x)

{

	if(S[x]!=x) S[x]=get(S[x]);

	return S[x];

}



int visit[N];



void dfs(int u,int pre)

{

	pos[u][0]=++id;

	mp[id]=u;

	int i;

	for(i=0;i<SZ(g[u]);i++)

	{

		int v=g[u][i];

		if(v==pre) continue;

		dfs(v,u);

		S[v]=u;

	}



	pos[u][1]=++id;

	mp[id]=u;



	visit[u]=1;

	for(i=0;i<SZ(Q[u]);i++)

	{

		int v=Q[u][i].first;

		int id=Q[u][i].second;

		if(visit[v]) b[id].lca=get(v);

	}

}



i64 ans[N][2];



int M;

int MM;



int cmp(node a,node b)

{

	if(a.x/MM!=b.x/MM) return a.x<b.x;



	if((a.x/MM)&1) return a.y>b.y; 



	return a.y<b.y;

}



int dp[N];





i64 A[N],B[N];



 

i64 suma(int x)

{

    i64 ans=0;

	while(x<N) ans+=A[x],x+=x&-x;

    return ans;

}

 

i64 sumb(int x)

{

	i64 ans=0;

	while(x) ans+=B[x],x-=x&-x;

    return ans;

}



i64 Ans;



int qp[N];



void add(int x)

{

	int j;

    for(j=d[mp[x]];j;j-=j&-j) A[j]+=qp[mp[x]];



    if(qp[mp[x]]==-1) Ans-=sumb(d[mp[x]]);





    for(j=d[mp[x]];j<=n;j+=j&-j)  B[j]+=(i64)dp[d[mp[x]]]*qp[mp[x]];



    Ans+=suma(d[mp[x]]+1)*dp[d[mp[x]]]*qp[mp[x]];



    if(qp[mp[x]]==1) Ans+=sumb(d[mp[x]]);

     

    qp[mp[x]]*=-1;

}



void deal()

{



	MM=sqrt(2*n);

	sort(b.begin(),b.end(),cmp);



	int L=1,R=0;

	int i;



	for(i=0;i<N;i++) qp[i]=1;



	for(i=0;i<SZ(b);i++)

	{

		int ll=b[i].x;

		int rr=b[i].y;

		int id=b[i].id;

		int t=b[i].t;



		while(R<rr) add(++R);

		while(R>rr) add(R--);

		while(L<ll) add(L++);

		while(L>ll) add(--L);



		if(b[i].flag) add(pos[b[i].lca][0]);

		ans[id][t]=Ans;

		if(b[i].flag) add(pos[b[i].lca][0]);

	}

}



int main()

{



	n=getInt();

	m=getInt();

	key=getInt();



	int i;

	for(i=1;i<=n;i++) d[i]=getInt(),dp[i]=d[i];

	sort(dp+1,dp+n+1);

	M=unique(dp+1,dp+n+1)-(dp+1);

	for(i=1;i<=n;i++) d[i]=lower_bound(dp+1,dp+M+1,d[i])-dp;



	int root;

	for(i=1;i<=n;i++)

	{

		int x=getInt();

		if(!x) root=i;

		else g[x].pb(i),g[i].pb(x);

	}

	int cnt=0;

	for(i=1;i<=m;i++)

	{

		char op[5];

		scanf("%s",op);

		if(op[0]=='C') root=getInt();

		else 

		{

			int x=getInt();

			node a;

			a.x=x%n+1;

			a.y=root;

			a.id=i;

			a.t=0;

			b.pb(a); Q[a.x].pb(MP(a.y,cnt)),Q[a.y].pb(MP(a.x,cnt)); cnt++;

		

			a.x=(x+key)%n+1;

			a.y=root;

			a.id=i;

			a.t=1;

			b.pb(a); Q[a.x].pb(MP(a.y,cnt)),Q[a.y].pb(MP(a.x,cnt)); cnt++;

		}

	}





	for(i=1;i<=n;i++) S[i]=i;

	dfs(1,0);





	for(i=0;i<SZ(b);i++)

	{

		if(b[i].y==b[i].lca) swap(b[i].x,b[i].y);

		if(b[i].x==b[i].lca)

		{

			b[i].flag=0;

			b[i].x=pos[b[i].x][0];

			b[i].y=pos[b[i].y][0];

			continue;

		}

		if(pos[b[i].x][1]>pos[b[i].y][0]) swap(b[i].x,b[i].y);

		b[i].x=pos[b[i].x][1];

		b[i].y=pos[b[i].y][0];

		b[i].flag=1;

	}



	clr(ans,-1);



	deal();



	int last=0;

	for(i=1;i<=m;i++) if(ans[i][0]!=-1||ans[i][1]!=-1)

	{

		printf("%lld\n",ans[i][last&1]);

		last=ans[i][last&1];

	}

}