主席树板子题练习

HDU:2665 k-th number
K-th number
一道主席树板题：
题解：
这是一道主席树的板子题，因为是在权值线段树的基础上写的主席树，所以需要进行离散化。主要利用了前缀和的思想，int root[ ] 这个数组记录动态开点，进行前缀求和操作即可

AC代码：

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
//主席树+离散化
const int maxn=1e6+20;
struct node
{
    int l,r,sum;
} tree[maxn*4];
vector<int>vec;
int id;

int root[maxn],a[maxn];
int getid(int x)
{
    return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}
void update(int l,int r,int &k,int x,int pos)
{
    tree[++id]=tree[x];
    tree[id].sum++;
    k=id;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(mid>=pos)
    {
        update(l,mid,tree[k].l,tree[x].l,pos);
    }
    else
    {
        update(mid+1,r,tree[k].r,tree[x].r,pos);
    }
}
int query(int l,int r,int k,int x,int ik)
{
    if(l==r)
        return l;
    int mid=(l+r)>>1;
    int sum=tree[tree[k].l].sum-tree[tree[x].l].sum;
    if(sum>=ik)
    {
        return query(l,mid,tree[k].l,tree[x].l,ik);
    }
    return query(mid+1,r,tree[k].r,tree[x].r,ik-sum);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m;
        id=0;
        scanf("%d %d",&n,&m);
        //for(int i=1; i<=n; i++)
            vec.clear();
        //memset(root,0,sizeof(root));
        //memset(tree,0,sizeof(tree));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            vec.push_back(a[i]);
        }
        sort(vec.begin(),vec.end());
        vec.erase(unique(vec.begin(),vec.end()),vec.end());
        int s,t,k;
        int cn=vec.size();
        for(int i=1; i<=n; i++)
        {
            update(1,cn,root[i],root[i-1],getid(a[i]));
        }
        while(m--)
        {
            scanf("%d %d %d",&s,&t,&k);
            int ans=query(1,cn,root[t],root[s-1],k)-1;
            printf("%d\n",vec[ans]);
        }
    }

}

POJ 2104
kth-number

两个题都测了一下，板子还是比较稳的

AC代码：

#include<algorithm>
#include<vector>
#include<cstdio>
using namespace std;
//主席树动态开点
const int maxn=1e5+7;
int a[maxn],root[maxn];//记录前一个树
int id;//动态节点
vector<int>vec;
struct node
{
    int l,r,sum;
} tree[maxn*40];
int getid(int x)//离散化
{
    return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}
void update(int l,int r,int &k,int x,int pos)
{
    tree[++id]=tree[x];//为x开一颗树
    tree[id].sum++;
    //tree[id].r=r;
    k=id;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(mid>=pos)
        update(l,mid,tree[k].l,tree[x].l,pos);
    else
        update(mid+1,r,tree[k].r,tree[x].r,pos);
}
int query(int l,int r,int k,int x,int ik)
{

    if(l==r)
    {
        return l;
    }
    int mid=(l+r)>>1;
    int sum=tree[tree[x].l].sum-tree[tree[k].l].sum;
    if(sum>=ik)
    {
        return query(l,mid,tree[k].l,tree[x].l,ik);
    }
    else
    {
        return query(mid+1,r,tree[k].r,tree[x].r,ik-sum);
    }
}

int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        vec.push_back(a[i]);
    }
    sort(vec.begin(),vec.end());
    vec.erase(unique(vec.begin(),vec.end()),vec.end());
    for(int i=1; i<=n; i++)
    {
        update(1,n,root[i],root[i-1],getid(a[i]));
    }
    while(m--)
    {
        int a,b,c;
        scanf("%d %d %d",&a,&b,&c);
        int ans=query(1,n,root[a-1],root[b],c)-1;
        printf("%d\n",vec[ans]);

    }

}