1. 程式人生 > >[Scoi2015] bzoj 4448 情報傳遞 - 主席樹

[Scoi2015] bzoj 4448 情報傳遞 - 主席樹

離線鏈上一維數點,主席樹即可
事實上可以繼續離線為BIT問題但沒啥意思就沒寫。
如果是離線鏈上二維數點不這麼做就麻煩了。

#include<bits/stdc++.h>
#define gc getchar()
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,v) rep(i,0,(int)v.size()-1)
#define lint long long
#define db double
#define pb push_back
#define mp make_pair
#define fir first
#define sec second #define N 200010 #define LOG 20 #define debug(x) cerr<<#x<<"="<<x #define sp <<" " #define ln <<endl using namespace std; typedef pair<int,int> pii; typedef set<int>::iterator sit; inline int inn() { int x,ch;while((ch=gc)<'0'||ch>'9'); x=
ch^'0';while((ch=gc)>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^'0');return x; } struct edges{ int to,pre; }e[N];int h[N],etop,n,x[N],y[N],c[N],opt[N],d[N],up[N][LOG],Log[N],tms[N]; inline int add_edge(int u,int v) { return e[++etop].to=v,e[etop].pre=h[u],h[u]=etop; } struct segment{
int s;segment *ch[2]; }*T[N]; int build(segment* &rt,int l,int r) { rt=new segment,rt->s=0;int mid=(l+r)>>1;if(l==r) return 0; build(rt->ch[0],l,mid),build(rt->ch[1],mid+1,r); return 0; } int update(segment* &x,segment* &y,int l,int r,int p) { x=new segment,x->ch[0]=y->ch[0],x->ch[1]=y->ch[1]; x->s=y->s+1;int mid=(l+r)>>1;if(l==r) return 0; if(p<=mid) update(x->ch[0],y->ch[0],l,mid,p); else update(x->ch[1],y->ch[1],mid+1,r,p); return 0; } int query(segment* &rt,int l,int r,int s,int t) { int mid=(l+r)>>1,ans=0;if(s<=l&&r<=t) return rt->s; if(s<=mid) ans+=query(rt->ch[0],l,mid,s,t); if(mid<t) ans+=query(rt->ch[1],mid+1,r,s,t); return ans; } int dfs(int x,int fa=0) { if(tms[x]<=n) update(T[x],T[fa],1,n,tms[x]);else T[x]=T[fa]; d[x]=d[up[x][0]=fa]+1;rep(i,1,Log[d[x]]) up[x][i]=up[up[x][i-1]][i-1]; for(int i=h[x];i;i=e[i].pre) dfs(e[i].to,x);return 0; } inline int getLCA(int x,int y) { if(d[x]<d[y]) swap(x,y); for(int i=Log[d[x]];i>=0;i--) if(d[up[x][i]]>=d[y]) x=up[x][i]; if(x==y) return x; for(int i=Log[d[x]];i>=0;i--) if(up[x][i]^up[y][i]) x=up[x][i],y=up[y][i]; return up[x][0]; } inline int query(int s,int t,int v) { int c=getLCA(s,t),v1=query(T[s],1,n,1,v),v2=query(T[t],1,n,1,v),v3=query(T[c],1,n,1,v); return !printf("%d %d\n",d[s]+d[t]-d[c]-d[c]+1,v1+v2-v3-(v3-(tms[c]<=v))); } int main() { n=inn();int rt=0;rep(i,1,n) tms[i]=n+1;build(T[0],1,n); rep(i,1,n) if(int fa=inn()) add_edge(fa,i);else rt=i; int q=inn();rep(i,2,n) Log[i]=Log[i>>1]+1; for(int i=1,z;i<=q;i++) if((opt[i]=inn())==2) z=inn(),tms[z]=++i; else x[i]=inn(),y[i]=inn(),c[i]=inn(); dfs(rt); rep(i,1,q) if(opt[i]==1) query(x[i],y[i],i-c[i]-1); return 0; }