BZOJ 1146 [CTSC2008]网络管理Network

题意：给一棵树点有点权，要求支持

1.更改点权

2.询问链上第k大

n,m<=80000

Sol:

并不能算Solution，因为我没过....

先树链剖分一发，询问变成给定log个不相交区间，询问区间并的第k大带单点修改

区间第k大用主席树，带单点修改就在外面套一层树状数组

这样的话时间复杂度是O(n*log^3(n) ) ，空间复杂度是O(n*log^2(n) )

看起来不错，写完调对交上即可得到MLE

空间限制162MB

经过一番压缩空间，最后压到164MB左右，GG

听说用指针/奇技淫巧可以再小一点，我太弱了不想做了

想到可以线段树套平衡树，询问时再套一层二分，以牺牲时间为代价换取空间

时间复杂度O(nlog^4(n) )，空间复杂度O(nlog(n) )

据说能过..不想写了

搞到数据发现第一个做法是能A的(不考虑空间)，我就在精神上A掉这个题好了QAQ

通过写数据结构题却A不掉，我们可以得到理性的愉悦和精神的放松，有理有据令人信服。

愿天堂没有MLE

Code:

// Memory_Limit_Exceed
#include<bits/stdc++.h>
#define debug(x) cout<<#x<<"="<<x<<endl
#define lb(x) x&(-x)
using namespace std;
const int maxn = 80001;
const int uuz = 14000000;
int first[maxn];
struct edg
{
	int next;
	int to;
}e[maxn<<1];
int e_sum;
int rt[maxn];
struct sg_tree
{
	int cnt,lson,rson;
}node[uuz];
int b[maxn<<1],val[maxn],L[1000],R[1000];
int fa[maxn],dep[maxn],siz[maxn],son[maxn],top[maxn],pos[maxn];
struct task{int opt,x,y;}q[maxn];
int n,m,tot,cnt;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void add_edg(int x,int y)
{
	e_sum++;
	e[e_sum].next=first[x];
	first[x]=e_sum;
	e[e_sum].to=y;
}
int find(int x)
{
	int l=1,r=b[0];
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(b[mid]==x) return mid;
		if(b[mid]<x) l=mid+1;else r=mid-1;
	}
}

void dfs1(int x,int f)
{
	fa[x]=f;dep[x]=dep[f]+1;siz[x]=1;
	for(int i=first[x];i;i=e[i].next)
	{
		if(e[i].to==f) continue;
		dfs1(e[i].to,x);
		siz[x]+=siz[e[i].to];
		if(siz[e[i].to]>siz[son[x]]) son[x]=e[i].to;
	}
}
void dfs2(int x,int t)
{
	top[x]=t;pos[x]=++cnt;
	if(son[x]) dfs2(son[x],t);
	for(int i=first[x];i;i=e[i].next)
	{
		if(e[i].to==fa[x]||e[i].to==son[x]) continue;
		dfs2(e[i].to,e[i].to);
	}
}

void insert(int &now,int last,int l,int r,int tar,int delta)
{
	now=++tot;
	node[now]=node[last];node[now].cnt+=delta;
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(tar<=mid) insert(node[now].lson,node[last].lson,l,mid,tar,delta);
	else insert(node[now].rson,node[last].rson,mid+1,r,tar,delta);
}
int query(int l,int r,int k)
{
	if(l==r) return l;
	int mid=(l+r)>>1,sum=0;
	for(int i=1;i<=L[0];i++) sum-=node[node[L[i]].rson].cnt;
	for(int i=1;i<=R[0];i++) sum+=node[node[R[i]].rson].cnt;
	if(sum>=k)
	{
		for(int i=1;i<=L[0];i++) L[i]=node[L[i]].rson;
		for(int i=1;i<=R[0];i++) R[i]=node[R[i]].rson;
		return query(mid+1,r,k);
	}
	else
	{
		for(int i=1;i<=L[0];i++) L[i]=node[L[i]].lson;
		for(int i=1;i<=R[0];i++) R[i]=node[R[i]].lson;
		return query(l,mid,k-sum);
	}
}

void add(int x,int y,int delta){for(;x<=n;x+=lb(x)) insert(rt[x],rt[x],1,b[0],y,delta);}
void ask(int l,int r)
{
	l--;
	for(;l;l-=lb(l)) L[++L[0]]=rt[l];
	for(;r;r-=lb(r)) R[++R[0]]=rt[r];
}

void solve(int x,int y,int k)
{
	L[0]=R[0]=0;
	int sum=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ask(pos[top[x]],pos[x]);
		sum+=(dep[x]-dep[top[x]]+1);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y]) swap(x,y);
	ask(pos[y],pos[x]);
	sum+=(dep[x]-dep[y]+1);
	if(sum<k) puts("invalid request!");
	else printf("%d\n",b[query(1,b[0],k)]);
}

int main()
{
	n=read();m=read();
	for(int i=1;i<=n;i++) b[++b[0]]=val[i]=read();
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read();
		add_edg(x,y);add_edg(y,x);
	}
	for(int i=1;i<=m;i++)
	{
		q[i].opt=read();q[i].x=read();q[i].y=read();
		if(q[i].opt==0) b[++b[0]]=q[i].y;
	}
	sort(b+1,b+1+b[0]);b[0]=unique(b+1,b+1+b[0])-b-1;
	for(int i=1;i<=n;i++) val[i]=find(val[i]);
	for(int i=1;i<=m;i++) if(!q[i].opt) q[i].y=find(q[i].y);
	dfs1(1,0);dfs2(1,1);
	for(int i=1;i<=n;i++) rt[i]=rt[i-1];
	for(int i=1;i<=n;i++) add(pos[i],val[i],1);
	for(int i=1;i<=m;i++)
	{
		if(q[i].opt==0)
		{
			int x=q[i].x;
			add(pos[x],val[x],-1);
			add(pos[x],q[i].y,1);
			val[x]=q[i].y;
		}
		else solve(q[i].x,q[i].y,q[i].opt);
	}
	return 0;
}