重开优快云，POJ2104

最新推荐文章于 2018-04-08 22:02:16 发布

路-过

最新推荐文章于 2018-04-08 22:02:16 发布

阅读量439

点赞数

本文链接：https://blog.youkuaiyun.com/u012896700/article/details/48654605

版权

本文介绍了一种名为主席树的数据结构，并提供了详细的实现代码。通过一个具体示例，讲解了如何初始化树、构建树以及查询过程。适合初学者了解主席树的基本原理及应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1年多没写优快云了。。

今天重新开始。

向学霸取经，知道了主席树的思路，第一次写主席树也不知道别人标准的代码是咋样的。。。

写的比较搓，继续开坑

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <bitset>
using namespace std;
#define   N  100000
struct tree1
{
    int left;
    int right;
    int zhi;
}tree[30*N];
int cnt=0;
int le[30*N];
int ri[30*N];
void init(int l,int r)
{
    tree[cnt].left=l;
    tree[cnt].right=r;
    tree[cnt].zhi=0;
    int node=cnt;
    cnt++;
    if(l==r) return ;
    int mid=(l+r)>>1;
    le[node]=cnt;
    init(l,mid);
    ri[node]=cnt;
    init(mid+1,r);
}
int id[N+5];
struct P
{
    int z,w;
}b[N+5];
bool flag;
bool cmp(P a1,P b1)
{
    if(!flag)
    return a1.z<b1.z;
    else return a1.w<b1.w;
}
int a[N+5];
void build_tree(int k)
{
    int re=id[k-1];
    id[k]=cnt;
    tree[cnt]=tree[re];
    tree[cnt].zhi++;
    int now=cnt;
    cnt++;
    while(tree[re].left!=tree[re].right)
    {
        int mid=(tree[re].left+tree[re].right)>>1;
        if(a[k-1]<=mid)
        {
            le[now]=cnt;
            ri[now]=ri[re];
            tree[cnt]=tree[le[re]];
            tree[cnt].zhi++;
            now=le[now];
            re=le[re];
            cnt++;
        }
        else
        {
            le[now]=le[re];
            ri[now]=cnt;
            tree[cnt]=tree[ri[re]];
            tree[cnt].zhi++;
            now=ri[now];
            re=ri[re];
            cnt++;
        }
    }
}
int query(int xb1,int xb2,int k)
{
    int l1=le[xb1];
    int l2=le[xb2];
    if(l1==-1&&l2==-1) return tree[xb1].left;
    if(tree[l2].zhi-tree[l1].zhi>=k) return query(l1,l2,k);
    else return query(ri[xb1],ri[xb2],k-(tree[l2].zhi-tree[l1].zhi));
}
int ans[N+5];
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=0;i<n;i++)
        {
            scanf("%d",&b[i].z);
            b[i].w=i;
        }
        memset(le,-1,sizeof(le));
        memset(ri,-1,sizeof(ri));
        flag=0;
        sort(b,b+n,cmp);
        for(int i=0;i<n;i++)
        {
            ans[i+1]=b[i].z;
            b[i].z=i+1;
        }
        flag=1;
        sort(b,b+n,cmp);
        for(int i=0;i<n;i++) a[i]=b[i].z;
        cnt=0;
        id[0]=0;
        init(1,n);
        for(int i=0;i<n;i++)
        {
            build_tree(i+1);
        }
        while(m--)
        {
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            int k1=query(id[l-1],id[r],k);
            printf("%d\n",ans[k1]);
        }
    }
}