【BZOJ1146】网络管理（主席树，树状数组）
【BZOJ1146】网络管理（主席树，树状数组）
题面
题解
树上带修改主席树
貌似和\(Count\ On\ A\ Tree\)那题很相似呀
只需要套上一个树状数组来维护修改好就好了
但是记住是用\(dfs\)来记录主席树的标号
一定不要搞错了
每一次修改只会影响他子数的值
而在\(dfs\)序上就是连续的一段
美滋滋的做完了
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 161000
inline int read()
{
RG int x=0,t=1;RG char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int dfn[MAX],low[MAX],fa[MAX],size[MAX],hson[MAX],dep[MAX],top[MAX];
int tim,n,Q;
int S[MAX<<1],Tot,TT[MAX];
int lowbit(int x){return x&(-x);}
struct Node{int ls,rs,v;}t[MAX<<8];
int rt[MAX],tot,ln[MAX];
int LCA(int u,int v)
{
while(top[u]!=top[v])
dep[top[u]]<dep[top[v]]?v=fa[top[v]]:u=fa[top[u]];
return dep[u]<dep[v]?u:v;
}
void Modify(int &now,int ff,int l,int r,int pos,int w)
{
t[now=++tot]=t[ff];t[now].v+=w;
if(l==r)return;
int mid=(l+r)>>1;
if(pos<=mid)Modify(t[now].ls,t[ff].ls,l,mid,pos,w);
else Modify(t[now].rs,t[ff].rs,mid+1,r,pos,w);
}
void PreModify(int K,int pos,int w){for(int i=K;i<=n;i+=lowbit(i))Modify(rt[i],rt[i],1,Tot,pos,w);}
int tmp1[MAX<<1],tmp2[MAX<<1],t1,t2;
int Query(int l,int r,int K)
{
if(l==r)return l;
int mid=(l+r)>>1,ss=0;
for(int i=1;i<=t1;++i)ss+=t[t[tmp1[i]].rs].v;
for(int i=1;i<=t2;++i)ss-=t[t[tmp2[i]].rs].v;
if(ss>=K)
{
for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].rs;
for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].rs;
return Query(mid+1,r,K);
}
else
{
for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].ls;
for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].ls;
return Query(l,mid,K-ss);
}
}
int preQuery(int u,int v,int K)
{
t1=t2=0;
for(int i=dfn[u];i;i-=lowbit(i))tmp1[++t1]=rt[i];
for(int i=dfn[v];i;i-=lowbit(i))tmp1[++t1]=rt[i];
int lca=LCA(u,v);
for(int i=dfn[lca];i;i-=lowbit(i))tmp2[++t2]=rt[i];
for(int i=dfn[fa[lca]];i;i-=lowbit(i))tmp2[++t2]=rt[i];
return Query(1,Tot,K);
}
void dfs1(int u,int ff)
{
size[u]=1;fa[u]=ff;dep[u]=dep[ff]+1;
for(int i=h[u];i;i=e[i].next)
{
if(e[i].v==ff)continue;
dfs1(e[i].v,u);
size[u]+=size[e[i].v];
if(size[e[i].v]>size[hson[u]])hson[u]=e[i].v;
}
}
void dfs2(int u,int tp)
{
top[u]=tp;dfn[u]=++tim;ln[tim]=u;
if(hson[u])dfs2(hson[u],tp);
for(int i=h[u];i;i=e[i].next)
{
if(e[i].v==fa[u]||e[i].v==hson[u])continue;
dfs2(e[i].v,e[i].v);
}
low[u]=tim;
}
void dfs3(int u,int ff)
{
PreModify(dfn[u],TT[u],1);
PreModify(low[u]+1,TT[u],-1);
for(int i=h[u];i;i=e[i].next)
if(e[i].v!=ff)dfs3(e[i].v,u);
}
struct Op{int k,a,b;}O[MAX];
int main()
{
n=read();Q=read();
for(int i=1;i<=n;++i)S[++Tot]=TT[i]=read();
for(int i=1;i<n;++i)
{
int u=read(),v=read();
Add(u,v);Add(v,u);
}
for(int i=1;i<=Q;++i)O[i].k=read(),O[i].a=read(),O[i].b=read();
for(int i=1;i<=Q;++i)if(O[i].k==0)S[++Tot]=O[i].b;
sort(&S[1],&S[Tot+1]);
Tot=unique(&S[1],&S[Tot+1])-S-1;
for(int i=1;i<=n;++i)TT[i]=lower_bound(&S[1],&S[Tot+1],TT[i])-S;
for(int i=1;i<=Q;++i)if(O[i].k==0)O[i].b=lower_bound(&S[1],&S[Tot+1],O[i].b)-S;
dfs1(1,0);dfs2(1,1);dfs3(1,1);
for(int i=1;i<=Q;++i)
{
if(O[i].k==0)
{
PreModify(dfn[O[i].a],TT[O[i].a],-1);
PreModify(low[O[i].a]+1,TT[O[i].a],1);
PreModify(dfn[O[i].a],O[i].b,1);
PreModify(low[O[i].a]+1,O[i].b,-1);
TT[O[i].a]=O[i].b;
}
else
{
int len=dep[O[i].a]+dep[O[i].b]-2*dep[LCA(O[i].a,O[i].b)]+1;
if(O[i].k>len){puts("invalid request!");continue;}
printf("%d\n",S[preQuery(O[i].a,O[i].b,O[i].k)]);
}
}
return 0;
}
