[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;
}

猜你喜欢

转载自blog.csdn.net/Mys_C_K/article/details/83148169