bzoj3545 [ONTAK2010]Peaks、bzoj3551 [ONTAK2010]Peaks加强版

题目描述：

bzoj3545，luogu

bzoj3551

题解：

重构树+线段树合并。

可以算是板子了吧。

代码（非强制在线）：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = ;
const int M = *N;
template<typename T>
inline void read(T&x)
{
    T f = ,c = ;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){c=c*+ch-'';ch=getchar();}
    x = f*c;
}
int n,m,q,to[N],h[N];
struct Pair
{
    int x,y;
}p[N];
bool cmp(Pair a,Pair b){return a.x<b.x;}
struct Edge
{
    int x,y,w;
    void rd(){read(x),read(y),read(w);}
}d[M];
bool emp(Edge a,Edge b){return a.w<b.w;}
int ff[N<<],fa[N<<][],wg[N<<],ch[N<<][];
int findff(int u){return u==ff[u]?u:ff[u]=findff(ff[u]);}
void kru()
{
    wg[] = 0x3f3f3f3f;
    for(int i=;i<=(n<<);i++)ff[i] = i;
    int tot = n,sum = ;
    for(int i=;i<=m&&sum<n-;i++)
    {
        int x = findff(d[i].x),y = findff(d[i].y);
        if(x!=y)
        {
            ff[x]=ff[y]=fa[x][]=fa[y][]=++tot;
            wg[tot]=d[i].w;
            sum++;
            ch[tot][]=x,ch[tot][]=y;
        }
    }
}
void init()
{
    for(int k=;(<<k)<=(n<<);k++)
        for(int i=;i<(n<<);i++)
            fa[i][k]=fa[fa[i][k-]][k-];
}
int rt[N<<];
struct segtree
{
    int siz[*N],ls[*N],rs[*N],tot;
    void insert(int l,int r,int&u,int qx)
    {
        if(!u)u=++tot;siz[u]++;
        if(l==r)return ;
        int mid = (l+r)>>;
        if(qx<=mid)insert(l,mid,ls[u],qx);
        else insert(mid+,r,rs[u],qx);
    }
    int merge(int x,int y)
    {
        if(!(x*y))return x+y;
        int z = ++tot;
        siz[z] = siz[x]+siz[y];
        ls[z] = merge(ls[x],ls[y]);
        rs[z] = merge(rs[x],rs[y]);
        return z;
    }
    int query(int l,int r,int u,int qk)
    {
        if(l==r)return l;
        int tmp = siz[rs[u]],mid = (l+r)>>;
        if(qk<=tmp)return query(mid+,r,rs[u],qk);
        else return query(l,mid,ls[u],qk-tmp);
    }
}tr;
int main()
{
//    freopen("tt.in","r",stdin);
    read(n),read(m),read(q);
    for(int i=;i<=n;i++)read(p[i].x),p[i].y=i;
    sort(p+,p++n,cmp);
    for(int las=0x3f3f3f3f,k=,i=;i<=n;i++)
    {
        if(las!=p[i].x)
        {
            las = p[i].x;
            to[++k] = las;
        }
        h[p[i].y] = k;
    }
    for(int i=;i<=m;i++)d[i].rd();
    sort(d+,d++m,emp);
    kru();init();
    for(int i=;i<=n;i++)tr.insert(,n,rt[i],h[i]);
    for(int i=n+;i<(n<<);i++)rt[i]=tr.merge(rt[ch[i][]],rt[ch[i][]]);
    for(int u,w,k,i=;i<=q;i++)
    {
        read(u),read(w),read(k);
        for(int j=;j>=;j--)if(wg[fa[u][j]]<=w)u=fa[u][j];
        if(tr.siz[rt[u]]<k)puts("-1");
        else printf("%d\n",to[tr.query(,n,rt[u],k)]);
    }
    return ;
}