标签：zoj text second 之间 com pre head tchar www
题目链接：开店
这道题显然一眼树分治，维护点分的结构，在每个点上，对每种年龄到这个点\(u\)以及他在点分树上父亲的距离和建一棵主席树，查询的时候一路往上跳即可。
但是我太懒了(其实你要说我不会也可以)，所以并不想写这种东西。于是，我就只能尝试一下别的方法。
设一个点\(u\)的年龄为\(y_u\)，\(u\)、\(v\)两点之间的距离为\(dis(u,v)\)，\(T_u=dis(root,u)\)，我们每次要求的式子是：
\begin{aligned} &\sum_{y_x\in [l,r]} dis(y,u)\\ =&\sum_{y_x\in [l,r]}(T_y+T_u-2T_{LCA(u,v)})\end{aligned}
注意到前面那两项我们是可以通过预处理前缀和\(O(1)\)求出的。于是我们就只需要考虑后面那坨东西怎么求。
我们可以考虑转化一下思路，转而求每条边的贡献。我们考虑对于一个点\(x\)满足\(y_x\in[l,r]\)，那么\(LCA(u,x)\)一直到根的路径都要被计算一次。那么我们就可以对于每个\(y_x\in[l,r]\)，把点\(x\)往上跳，途中经过的边标记加\(1\)。那么最后我们再从\(u\)往上跳，每条边的的标记数就是这条边被计算的次数。那么我们就只需要快速维护一个点到根的路径即可。这个可以树链剖分之后用权值线段树解决。
下面贴代码：
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
#define maxn 150010
#define MAXN 6000010
#define point pair<int,int>
#define sa first
#define sb second
using namespace std;
typedef long long llg;
point s[maxn];
int n,Q,A,a[maxn],da[maxn],ld,dep[maxn],c1[maxn];//c1[x]表示[1,x]出现的次数和
int head[maxn],next[maxn<<1],to[maxn<<1],c[maxn<<1],tt;//邻接表
int fa[maxn],top[maxn],siz[maxn],son[maxn],tc[maxn],wd[maxn],fd[maxn];//树链剖分
int rt[maxn],addv[MAXN],le[MAXN],ri[MAXN],L,R,CO,_1,_2;//主席树
llg ans,c2[maxn],sumv[MAXN],ji;//c2[x]表示∑dep[u](a[u]∈[1,x])
int getint(){
int w=0;bool q=0;
char c=getchar();
while((c>‘9‘||c<‘0‘)&&c!=‘-‘) c=getchar();
if(c==‘-‘) c=getchar(),q=1;
while(c>=‘0‘&&c<=‘9‘) w=w*10+c-‘0‘,c=getchar();
return q?-w:w;
}
void link(int x,int y){
to[++tt]=y;next[tt]=head[x];head[x]=tt;
to[++tt]=x;next[tt]=head[y];head[y]=tt;
c[tt]=c[tt-1]=getint();
}
void dfs(int u){
siz[u]=1; c1[a[u]]++; c2[a[u]]+=dep[u];
for(int i=head[u],v;v=to[i],i;i=next[i])
if(!siz[v]){
dep[v]=dep[u]+c[i];
fa[v]=u; dfs(v); siz[u]+=siz[v];
if(siz[v]>siz[son[u]]) son[u]=v;
}
}
void dfs(int u,int ot){
top[u]=ot; tc[u]=++tt;
wd[tt]=dep[u]; fd[tt]=dep[fa[u]];
if(son[u]) dfs(son[u],ot);
for(int i=head[u],v;v=to[i],i;i=next[i])
if(!top[v]) dfs(v,v);
}
void add(int &u,int l,int r){
tt++; addv[tt]=addv[u];
le[tt]=le[u],ri[tt]=ri[u];
sumv[tt]=sumv[u]; u=tt;
int mid=(l+r)>>1;
if(l>=L && r<=R){
sumv[u]+=wd[r]-fd[l];
addv[u]++; return;
}
if(L<=mid) add(le[u],l,mid);
if(R>mid) add(ri[u],mid+1,r);
sumv[u]=sumv[le[u]]+sumv[ri[u]];
sumv[u]+=(llg)addv[u]*(wd[r]-fd[l]);
}
void work(int u,int co){
while(u){
L=tc[top[u]],R=tc[u];
add(rt[co],1,n); u=fa[top[u]];
}
}
void query(int u1,int u2,int l,int r){
int mid=(l+r)>>1;
if(l>=L && r<=R){
ji+=sumv[u2]+(llg)_2*(wd[r]-fd[l]);
ji-=sumv[u1]+(llg)_1*(wd[r]-fd[l]);
return;
}
_1+=addv[u1]; _2+=addv[u2];
if(L<=mid) query(le[u1],le[u2],l,mid);
if(R>mid) query(ri[u1],ri[u2],mid+1,r);
_1-=addv[u1]; _2-=addv[u2];
}
int up(int x){//二分>=x的第一个
int l=1,r=ld,mid;
while(l!=r){
mid=(l+r)>>1;
if(da[mid]>=x) r=mid;
else l=mid+1;
}
return l;
}
int lo(int x){//二分<=x的第一个
int l=1,r=ld,mid;
while(l!=r){
mid=(l+r+1)>>1;
if(da[mid]<=x) l=mid;
else r=mid-1;
}
return l;
}
int main(){
File("shop");
n=getint(); Q=getint(); A=getint(); ld=n;
for(int i=1;i<=n;i++) a[i]=da[i]=getint(); da[++ld]=A+1;
sort(da+1,da+ld+1); ld=unique(da+1,da+ld+1)-da-1;
for(int i=1;i<=n;i++) a[i]=up(a[i]),s[i]=make_pair(a[i],i);
for(int i=1;i<n;i++) link(getint(),getint());
tt=0; dfs(1); dfs(1,1); sort(s+1,s+n+1); tt=0;
for(int i=1;i<=ld;i++) c1[i]+=c1[i-1],c2[i]+=c2[i-1];
for(int i=1;i<=n;i++){
if(s[i].sa!=s[i-1].sa) rt[s[i].sa]=rt[s[i-1].sa];
CO=s[i].sa; work(s[i].sb,s[i].sa);
}
rt[ld]=rt[ld-1];
while(Q--){
int u=getint(),aa=getint(),bb=getint(),l,r; ji=0;
(aa+=ans%A)%=A; (bb+=ans%A)%=A;
l=min(aa,bb),r=max(aa,bb); l=up(l); r=lo(r);
if(l>r) ans=ji=0;
else{
ans=c2[r]-c2[l-1]+(llg)(c1[r]-c1[l-1])*dep[u];
while(u){
L=tc[top[u]],R=tc[u];
query(rt[l-1],rt[r],1,n);
u=fa[top[u]];
}
ans-=ji<<1;
}
printf("%lld\n",ans);
}
return 0;
}
标签：zoj text second 之间 com pre head tchar www
原文地址：http://www.cnblogs.com/lcf-2000/p/6339396.html
