3545: [ONTAK2010]Peaks

题目链接

题目大意：给定一个无向图，每个点和每条边都有权值，多次询问从点v开始只能经过边权小于等于x的点中权值第k大

题解：离线，把边和询问都按照边权从小到大排序，动态加边
对每个点维护一颗权值线段树，每次计算答案之前将边权小于等于限制的边两端的连通块（并查集维护）的权值线段树合并，然后在权值线段树上查询即可

因为用了权值线段树，要离散化h

我的收获：线段树合并吼啊

#include <iostream>#include <cstdio>#include <cstring>#include <cstdlib>#include <algorithm>using namespace std;#define M 100005#define MM 500005int n,m,q,cnt;int h[M],z[M],f[M],root[M],ans[MM];int tl[MM*10],tr[MM*10],sum[MM*10];struct data{int x,y,w,ans,id,f;}a[MM*2];bool operator<(data a,data b){return a.w==b.w?a.f<b.f:a.w<b.w;}inline int find(int x){return x==f[x]?x:f[x]=find(f[x]);}inline void pushup(int x){sum[x]=sum[tl[x]]+sum[tr[x]];}void build(int k,int l,int r,int &x){    if(!x) x=++cnt;sum[x]=1;//动态开点     if(l==r) return ;    int m=(l+r)>>1;    if(k<=m) build(k,l,m,tl[x]);    else build(k,m+1,r,tr[x]);}int query(int w,int l,int r,int x)//求排名为w的值，类似平衡树 {    if(l==r) return l;    int m=(l+r)>>1;    if(sum[tl[x]]>=w) return query(w,l,m,tl[x]);    else return query(w-sum[tl[x]],m+1,r,tr[x]);}int merge(int x,int y){    if(!x||!y) return x+y;    sum[x]+=sum[y];    tl[x]=merge(tl[x],tl[y]);    tr[x]=merge(tr[x],tr[y]);    return x;}void solve(){    for(int i=1;i<=n;i++) build(h[i],1,n,root[i]);    for(int i=1;i<=m+q;i++)    if(!a[i].f){        int fx=find(a[i].x),fy=find(a[i].y);        if(fx!=fy) f[fx]=fy,root[fy]=merge(root[fx],root[fy]);    }    else{        int fx=find(a[i].x);        if(sum[root[fx]]<a[i].y) ans[a[i].id]=-1;//tan90°         else ans[a[i].id]=z[query(sum[root[fx]]-a[i].y+1,1,n,root[fx])];//把第k大转化为第sum-k+1小，离散化了要还原     }}void work(){    solve();    for(int i=1;i<=q;i++) printf("%d\n",ans[i]);}void init(){    cin>>n>>m>>q;    for(int i=1;i<=n;i++) scanf("%d",&h[i]),z[i]=h[i],f[i]=i;    sort(z+1,z+n+1);    for(int i=1;i<=n;i++) h[i]=lower_bound(z+1,z+n+1,h[i])-z;    for(int i=1;i<=m;i++) a[i].f=0,scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].w);    for(int i=m+1;i<=m+q;i++) a[i].f=1,a[i].id=i-m,scanf("%d%d%d",&a[i].x,&a[i].w,&a[i].y);    sort(a+1,a+m+q+1);}int main(){    init();    work();    return 0;}