重开CSDN,POJ2104

1年多没写CSDN了。。

今天重新开始。

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

写的比较搓,继续开坑


#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]);
        }
    }
}


评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值