首页 > 代码库 > BZOJ 4012 【HNOI2015】 开店
BZOJ 4012 【HNOI2015】 开店
题目链接:开店
这道题显然一眼树分治,维护点分的结构,在每个点上,对每种年龄到这个点\(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; }
BZOJ 4012 【HNOI2015】 开店
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。