[主席树 树状数组套权值线段树] BZOJ 1146 [CTSC2008]网络管理Network

题目大意：带修改树上路径第k大

CTSC的裸题么

先建主席树 

然后修改么 可以发现对整棵子树有影响 对每个点记录每个权值的改变量

然后用差分的思想 用树状数组区间修改 单点查询

A掉后发现不用打主席树 可以最初的权值也可以树状数组上改 

结果稍微改下程序 慢了5倍233

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define V G[p].v
using namespace std;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x)
{
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=80005;
const int M=8000005;

int sx[N*2],icnt;
inline int Bin(int x) { return lower_bound(sx+1,sx+icnt+1,x)-sx; }

int Stk[M],pnt;
inline void Init() { pnt=0; for (int i=M-1;i;i--) Stk[++pnt]=i; }
inline int New() { return Stk[pnt--]; }
inline void Del(int x) { Stk[++pnt]=x; }

int n,val[N];
int root[N],ls[M],rs[M],sum[M];
bool notnull[M];

void Modify(int &y,int x,int l,int r,int t,int f){
	y=New(); int mid=(l+r)>>1;
	if (l==r) { sum[y]=sum[x]+f; return; }
	if (t<=mid)
		rs[y]=rs[x],Modify(ls[y],ls[x],l,mid,t,f);
	else
		ls[y]=ls[x],Modify(rs[y],rs[x],mid+1,r,t,f);
 	sum[y]=sum[ls[y]]+sum[rs[y]];
}

void Add(int &x,int l,int r,int t,int f){
	if (!x) x=New(); int mid=(l+r)>>1;
	if (l==r)
		sum[x]+=f,(sum[x]?notnull[x]=1:notnull[x]=0);
	else if (t<=mid)
		Add(ls[x],l,mid,t,f),sum[x]=sum[ls[x]]+sum[rs[x]],notnull[x]=notnull[ls[x]]||notnull[rs[x]];
	else
		Add(rs[x],mid+1,r,t,f),sum[x]=sum[ls[x]]+sum[rs[x]],notnull[x]=notnull[ls[x]]||notnull[rs[x]];
	if (!notnull[x]) Del(x),x=0;
}

namespace BIT{
	#define lowbit(x) ((x)&-(x))
	int c[N],maxn;
	inline void init(int n){
		maxn=n;
	}
	inline void add(int x,int t,int f){
		for (int i=x;i<=maxn;i+=lowbit(i))
			Add(c[i],1,icnt,t,f);
	}
	inline void add(int l,int r,int t,int f){
		add(l,t,f); add(r+1,t,-f);
	}
	inline void divi(int x,int *a,int &tot){
		tot=0;
		for (int i=x;i;i-=lowbit(i))
			if (c[i]) a[++tot]=c[i];
	}
}

struct edge{ int u,v,next; }G[N*2];
int head[N],inum;
inline void addedge(int u,int v,int p){
	G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int clk;
int fat[N][21],depth[N],tid[N],size[N];

inline void dfs(int u,int fa){
	size[u]=1; tid[u]=++clk; fat[u][0]=fa; depth[u]=depth[fa]+1;
	for (int k=1;k<=20;k++) fat[u][k]=fat[fat[u][k-1]][k-1];
//	Modify(root[u],root[fa],1,icnt,val[u]=Bin(val[u]),1);
	for (int p=head[u];p;p=G[p].next) if (V!=fa) dfs(V,u),size[u]+=size[V];
}

inline int LCA(int u,int v){
	if (depth[u]<depth[v]) swap(u,v);
	for (int k=20;~k;k--) if ((depth[u]-depth[v])&(1<<k)) u=fat[u][k]; if (u==v) return u;
	for (int k=20;~k;k--) if (fat[u][k]!=fat[v][k]) u=fat[u][k],v=fat[v][k]; return fat[u][0];
}

struct abcd{ int k,a,b; }eve[N];
int Q;

int a[1005],b[1005],c[1005],d[1005];
int atot,btot,ctot,dtot;

inline int Query(int l,int r,int k){
	int mid=(l+r)>>1,tmp=0;
	if (l==r) return sx[l];
	for (int i=0;i<=atot;i++) tmp+=sum[rs[a[i]]];
	for (int i=0;i<=btot;i++) tmp+=sum[rs[b[i]]];
	for (int i=0;i<=ctot;i++) tmp-=sum[rs[c[i]]];
	for (int i=0;i<=dtot;i++) tmp-=sum[rs[d[i]]];
	if (k<=tmp)
	{
		for (int i=0;i<=atot;i++) a[i]=rs[a[i]];
		for (int i=0;i<=btot;i++) b[i]=rs[b[i]];
		for (int i=0;i<=ctot;i++) c[i]=rs[c[i]];
		for (int i=0;i<=dtot;i++) d[i]=rs[d[i]];
		return Query(mid+1,r,k);
	}
	else
	{
		for (int i=0;i<=atot;i++) a[i]=ls[a[i]];
		for (int i=0;i<=btot;i++) b[i]=ls[b[i]];
		for (int i=0;i<=ctot;i++) c[i]=ls[c[i]];
		for (int i=0;i<=dtot;i++) d[i]=ls[d[i]];
		return Query(l,mid,k-tmp);
	}	
}

inline void Schange(int u,int x,int f)
{
	int st=tid[u],ed=tid[u]+size[u]-1;
	if (f) BIT::add(st,ed,val[u],-1);
	val[u]=x;
	BIT::add(st,ed,val[u],1);
}

inline void Squery(int u,int v,int k)
{
	int lca=LCA(u,v),ret;
	a[0]=root[u]; 			BIT::divi(tid[u],a,atot);
	b[0]=root[v]; 			BIT::divi(tid[v],b,btot);
	c[0]=root[lca]; 		BIT::divi(tid[lca],c,ctot);
	d[0]=root[fat[lca][0]]; BIT::divi(tid[fat[lca][0]],d,dtot);
	int tmp=0;
	for (int i=0;i<=atot;i++) tmp+=sum[a[i]];
	for (int i=0;i<=btot;i++) tmp+=sum[b[i]];
	for (int i=0;i<=ctot;i++) tmp-=sum[c[i]];
	for (int i=0;i<=dtot;i++) tmp-=sum[d[i]];
	if (k<=tmp)
		ret=Query(1,icnt,k);
	else
		ret=-1;
	ret==-1?printf("invalid request!\n"):printf("%d\n",ret);
}

int main()
{
	int iu,iv;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	Init();
	read(n); read(Q);
	BIT::init(n);
	for (int i=1;i<=n;i++) read(val[i]),sx[++icnt]=val[i];
	for (int i=1;i<n;i++) read(iu),read(iv),addedge(iu,iv,++inum),addedge(iv,iu,++inum);
	for (int i=1;i<=Q;i++)
		read(eve[i].k),read(eve[i].a),read(eve[i].b),(!eve[i].k?sx[++icnt]=eve[i].b:0);
	sort(sx+1,sx+icnt+1); icnt=unique(sx+1,sx+icnt+1)-sx-1;
	dfs(1,0);
	for (int i=1;i<=n;i++)
		Schange(i,Bin(val[i]),0);
	for (int i=1;i<=Q;i++)
		if (!eve[i].k)
			Schange(eve[i].a,eve[i].b=Bin(eve[i].b),1);
		else
			Squery(eve[i].a,eve[i].b,eve[i].k);
	return 0;
}