题意:给一棵树点有点权,要求支持
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;
}