1. 程式人生 > >BZOJ 3626 [LNOI2014]LCA

BZOJ 3626 [LNOI2014]LCA

str alt print fine 一個點 hid 答案 -s ring

一道比較神奇的題。

樹鏈剖分+奇技淫巧;

神奇地發現,把z到跟的路徑上的點值+1,查詢一個點到跟的路徑和就是它與z的lca的深度。

相對的,把l~r到跟的路徑上的點值+1,查詢z到跟的路徑和就是要的答案。

考慮差分,把一個詢問拆成兩個,把所有詢問排序然後從0~n-1到跟路徑上的值+1;

一開始狂WA,發現把線段樹區間加的(l-r)*v打成了(qr-ql)*v了。。。

技術分享
//Twenty
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include
<cstring> #include<queue> #include<vector> typedef long long LL; using namespace std; const int mod=201314; const int N=200000+299; int n,m,cnt,l,r,z; LL ans[N]; struct qs{ int id,x,f,ts,z; qs(){} qs(int id,int x,int f,int ts,int z):id(id),x(x),f(f),ts(ts),z(z){} friend bool
operator <(const qs&a,const qs&b) { return a.x<b.x; } }q[N]; int ecnt,fir[N],nxt[N],to[N]; void add(int u,int v) { nxt[++ecnt]=fir[u]; fir[u]=ecnt; to[ecnt]=v; } #define lc x<<1 #define rc x<<1|1 #define mid ((l+r)>>1) LL sg[N<<2],lz[N<<2
]; void down(int x,int l_len,int r_len) { if(!lz[x]) return; if(lc) (sg[lc]+=l_len*lz[x])%=mod,(lz[lc]+=lz[x])%=mod; if(rc) (sg[rc]+=r_len*lz[x])%=mod,(lz[rc]+=lz[x])%=mod; lz[x]=0; } void add(int x,int l,int r,int ql,int qr,LL v) { if(l>=ql&&r<=qr) { (sg[x]+=v*(r-l+1))%=mod; (lz[x]+=v)%=mod; return;} down(x,mid-l+1,r-mid); if(ql<=mid) add(lc,l,mid,ql,qr,v); if(qr>mid) add(rc,mid+1,r,ql,qr,v); sg[x]=sg[lc]+sg[rc]; if(sg[x]>=mod) sg[x]-=mod; } LL query(int x,int l,int r,int ql,int qr) { if(l>=ql&&r<=qr) return sg[x]; down(x,mid-l+1,r-mid); if(qr<=mid) return query(lc,l,mid,ql,qr); if(ql>mid) return query(rc,mid+1,r,ql,qr); LL res=query(lc,l,mid,ql,qr)+query(rc,mid+1,r,ql,qr); return res>=mod?res-mod:res; } int R[N],sz[N]; void DFS(int x) { sz[x]=1; for(int i=fir[x];i;i=nxt[i]) { R[to[i]]=R[x]+1; DFS(to[i]); sz[x]+=sz[to[i]]; } } int tot,top[N],fa[N],tid[N]; void dfs(int x,int t) { top[x]=t; tid[x]=++tot; int mson=0; for(int i=fir[x];i;i=nxt[i]) if(!mson||sz[to[i]]>=sz[mson]) mson=to[i]; if(!mson) return; dfs(mson,t); for(int i=fir[x];i;i=nxt[i]) if(to[i]!=mson) dfs(to[i],to[i]); } void schange(int l,int r,int v) { while(top[l]!=top[r]) { if(R[top[l]]<R[top[r]]) swap(l,r); add(1,1,n,tid[top[l]],tid[l],v); l=fa[top[l]]; } if(tid[l]>tid[r]) swap(l,r); add(1,1,n,tid[l],tid[r],v); } LL squery(int l,int r) { LL res=0; while(top[l]!=top[r]) { if(R[top[l]]<R[top[r]]) swap(l,r); res+=query(1,1,n,tid[top[l]],tid[l]); if(res>=mod) res-=mod; l=fa[top[l]]; } if(tid[l]>tid[r]) swap(l,r); res+=query(1,1,n,tid[l],tid[r]); if(res>=mod) res-=mod; return res; } int main() { scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { scanf("%d",&fa[i]); add(fa[i],i); } DFS(0); dfs(0,0); for(int i=1;i<=m;i++) { scanf("%d%d%d",&l,&r,&z); if(l) q[++cnt]=qs(i,l-1,-1,0,z); q[++cnt]=qs(i,r,1,0,z); } sort(q+1,q+cnt+1); int now=1; for(int i=0;i<n;i++) { schange(0,i,1); while(now<=m*2&&q[now].x<=i) { LL res=squery(0,q[now].z); ans[q[now].id]+=q[now].f*res; now++; if(now>2*m) break; } } for(int i=1;i<=m;i++) printf("%lld\n",(ans[i]+mod)%mod); return 0; }
View Code

BZOJ 3626 [LNOI2014]LCA