12/4周总结

周末返乡，刚下高铁，白天再编...

总的来说，这周过的非常的魔幻，周四以前还一切正常的学习和生活，周四晚上收到学校提前放假的消息，就开始各种抢票啥的，确实影响了学习的心态，心不在焉的，本来这周学习主要就是在着重复习14周考试科目，没有学习什么新的知识，多看了看线段树和主席树的题，CDQ对我来说还是有点困难，这些题都是需要沉下心来思考的。这周估计也学不了多少，得新冠真的太难受。。

贴两道题吧

1）这道题就是区间查询第k大元素，符合主席树的特征，采用非递归的写法,

#include <bits/stdc++.h>
using namespace std;
const int maxn=100005;
int n,m;
int v[maxn],b[maxn],t[maxn],tsize,tot=0;
struct charitree
{
    int l,r,tsize;
}a[maxn*30];
int build(int l,int r)
{
    int root=tot++;
    a[root].tsize=0;
    if(l==r)
        return root;
    int m=(l+r)>>1;
    a[root].l=build(l,m);
    a[root].r=build(m+1,r);
    return root;
}
int update(int root,int x)
{
    int now=tot++;
    int tmp=now;
    a[now].tsize=a[root].tsize+1;
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(x<=m)
        {
            a[now].l=tot++;
            a[now].r=a[root].r;
            now=a[now].l;
            r=m;
            root=a[root].l;
        }
        else
        {
            a[now].l=a[root].l;
            a[now].r=tot++;
            now=a[now].r;
            l=m+1;
            root=a[root].r;
        }
        a[now].tsize=a[root].tsize+1;
    }
    return tmp;
}
int ask(int lx,int rx,int k)
{
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(a[a[rx].l].tsize-a[a[lx].l].tsize>=k)
        {
            r=m;
            lx=a[lx].l;
            rx=a[rx].l;
        }
        else
        {
            l=m+1;
            k-=a[a[rx].l].tsize-a[a[lx].l].tsize;
            lx=a[lx].r;
            rx=a[rx].r;
        }
    }
    return l;
}
void hash1()
{
    sort(b+1,b+1+n);
    tsize=unique(b+1,b+1+n)-b-1;
}
int find_index(int x)
{
    return lower_bound(b+1,b+1+tsize,x)-b;
}
int main()
{
    int i,j,l,r,k;
    cin>>n>>m;
    for(i=1;i<=n;i++)
        cout<<v[i];b[i]=v[i];
    hash1();
    t[0]=build(1,tsize);
    for(i=1;i<=n;i++)
        t[i]=update(t[i-1],find_index(v[i]));
    while(m--)
    {
        cin>>l>>r>>k;
        cout<<b[ask(t[l-1],t[r],k)]<<endl;
    }
    return 0;
}

其实有点模版的样子，对于同类型找区间最小值，只是需要修改一下查询的函数就可以了。

---

2）这种题也是用主席树，和上题只是线性的建树和树上建树的区别，每一个节点由其父节点修改而来，两个节点的路径lca求一求就可以。由于是只是查询，所以可以在线。大概题意就是给定了一颗树，每个节点都有一个权值，查询两个节点之间路径上的权值第k小节点。

#include<bits/stdc++.h>

using namespace std;
const int maxn=100005;
int n,m;
int v[maxn],b[maxn],t[maxn],tsize,tot=0;
struct charitree
{
    int l,r,tsize;
}a[maxn*30];
vector<int>w[maxn];
 
int book[maxn],fa[maxn],dep[maxn],up[maxn][20];
int build(int l,int r)
{
    int root=tot++;
    a[root].tsize=0;
    if(l==r)
        return root;
    int m=(l+r)>>1;
    a[root].l=build(l,m);
    a[root].r=build(m+1,r);
    return root;
}
int update(int root,int x)
{
    int now=tot++;
    int tmp=now;
    a[now].tsize=a[root].tsize+1;
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(x<=m)
        {
            a[now].l=tot++;
            a[now].r=a[root].r;
            now=a[now].l;
            r=m;
            root=a[root].l;
        }
        else
        {
            a[now].l=a[root].l;
            a[now].r=tot++;
            now=a[now].r;
            l=m+1;
            root=a[root].r;
        }
        a[now].tsize=a[root].tsize+1;
    }
    return tmp;
}
void hash1()
{
    sort(b+1,b+1+n);
    tsize=unique(b+1,b+1+n)-b-1;
}
int find_index(int x)
{
    return lower_bound(b+1,b+1+tsize,x)-b;
}
int init()
{
    memset(up,0,sizeof(up));
    for(int i=1;i<=n;i++)
        up[i][0]=fa[i];
    for(int j=1;j<=16;j++)
    {
        for(int i=1;i<=n;i++)
        {
            up[i][j]=up[up[i][j-1]][j-1];
        }
    }
}
int lca(int lx,int rx)
{
    if(dep[lx]<dep[rx])
        swap(lx,rx);
    int a=rx,b=lx;
    int c=dep[b]-dep[a];
    for(int i=0;i<=16;i++)
    {
        if(c&(1<<i))
        b=up[b][i];
    }
    if(a==b)
        return a;
    for(int i=16;i>=0;i--)
    {
        if(up[a][i]!=up[b][i])
        {
            a=up[a][i];
            b=up[b][i];
        }
    }
    return fa[a];
}
int ask(int lx,int rx,int k)
{
    int l=1,r=tsize;
    int mx=lca(lx,rx);
    int f=fa[mx];
    lx=t[lx];
    rx=t[rx];
    mx=t[mx];
    f=t[f];
    while(l<r)
    {
        int m=(l+r)>>1;
   //     cout<<b[m]<<" "<<a[a[rx].l].tsize<<" "<<a[a[f].l].tsize<<" "<<a[a[lx].l].tsize<<" "<<a[a[mx].l].tsize<<endl;
    //    cout<<b[m]<<" "<<a[a[rx].r].tsize<<" "<<a[a[f].r].tsize<<" "<<a[a[lx].r].tsize<<" "<<a[a[mx].r].tsize<<endl;
        if(a[a[rx].l].tsize-a[a[f].l].tsize+a[a[lx].l].tsize-a[a[mx].l].tsize>=k)
        {
            r=m;
            lx=a[lx].l;
            f=a[f].l;
            mx=a[mx].l;
            rx=a[rx].l;
        }
        else
        {
            l=m+1;
            k-=a[a[rx].l].tsize-a[a[f].l].tsize+a[a[lx].l].tsize-a[a[mx].l].tsize;
            lx=a[lx].r;
            f=a[f].r;
            mx=a[mx].r;
            rx=a[rx].r;
        }
    }
    return l;
}
void dfs(int x)
{
    for(int i=0;i<w[x].size();i++)
    {
        if(book[w[x][i]])
            continue;
        book[w[x][i]]=1;
        dep[w[x][i]]=dep[x]+1;
        fa[w[x][i]]=x;
        t[w[x][i]]=update(t[x],find_index(v[w[x][i]]));
        dfs(w[x][i]);
    }
    return ;
}
int main()
{
    int i,j,l,r,k;
    cin>>n>>m;
    for(i=1;i<=n;i++)
        cin>>v[i],b[i]=v[i];
    for(i=1;i<n;i++)
    {
        cin>>l>>r;
        w[l].push_back(r);
        w[r].push_back(l);
    }
    hash1();
    t[0]=build(1,tsize);
    int root=1;
    w[0].push_back(root);
    book[0]=1;dep[0]=0;
    dfs(0);
    init();
    while(m--)
    {
        cin>>l>>r>>k;
        cout<<b[ask(l,r,k)]<<endl;
    }
    return 0;
}