HDU 2665 Kth number 题解&代码

嘛…作为一道主席树模板题我TLE RE MLE占全然后卡到998ms过也是蛮拼的【事后删了一个无用的memset然后904ms卡过去了…好歹不那么危险了
其实这题是划分树= =毕竟静态查询…主席树肯定慢，加上姿势问题…想不T也挺艰辛的
和BZOJ 3207并没有什么卵区别，甚至少了hash过程= =
建n个线段树，查询第s-1到第t个树的size大小，和k比较即可

20min后：
不用map排序变成了717ms…嗯STL这种奇怪的物种窝一定小心
新代码在最下

Old Accept 904ms

#include<iostream>
#include<algorithm>
#include<map>
#include<string.h>
#include<stdio.h>
using namespace std;
const int maxn=100005;
int T,m,n,L,R,K,lim,tot,cnt,line[maxn],p[maxn],fp[maxn];
int tree[maxn*20],son[2][maxn*20],rt[maxn];
map<int,int> vis;
void init(void)
{
    lim=1;tot=0;cnt=0;
    memset(tree,0,sizeof(tree));
    memset(son,0,sizeof(son));
    vis.clear();
}
void add_tree(int rl,int rn,int l,int r,int value)
{
    tree[rn]=tree[rl]+1;
    if(l==r)return;
    int mid=(l+r)/2;
    if(value>mid)son[0][rn]=son[0][rl],son[1][rn]=++tot,add_tree(son[1][rl],son[1][rn],mid+1,r,value);
    else son[1][rn]=son[1][rl],son[0][rn]=++tot,add_tree(son[0][rl],son[0][rn],l,mid,value);
}
int query(int rl,int rn,int l,int r,int k)
{
    if(l==r)return fp[l];
    int mid=(l+r)/2;
    if(k>tree[son[0][rn]]-tree[son[0][rl]])return query(son[1][rl],son[1][rn],mid+1,r,k-tree[son[0][rn]]+tree[son[0][rl]]);
    else return query(son[0][rl],son[0][rn],l,mid,k);
}
bool cmp(int a,int b)
{
    return line[a]<line[b];
}
int main(void)
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&line[i]);
            p[i]=i;
            lim=max(lim,line[i]);
        }
        sort(p+1,p+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
            vis[line[p[i]]]=++cnt;
            fp[cnt]=line[p[i]];
        }
        for(int i=1;i<=n;i++)
        {
            rt[i]=++tot;
            add_tree(rt[i-1],rt[i],1,cnt,vis[line[i]]);
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&L,&R,&K);
            printf("%d\n",query(rt[L-1],rt[R],1,cnt,K));
        }
    }
    return 0;
}

NEW Accept 717ms

#include<iostream>
#include<algorithm>
#include<map>
#include<string.h>
#include<stdio.h>
using namespace std;
const int maxn=100005;
int T,m,n,L,R,K,lim,tot,cnt,line[maxn],vis[maxn],p[maxn],fp[maxn];
int tree[maxn*20],son[2][maxn*20],rt[maxn];
void init(void)
{
    lim=1;tot=0;cnt=0;
    memset(tree,0,sizeof(tree));
    memset(son,0,sizeof(son));
}
void add_tree(int rl,int rn,int l,int r,int value)
{
    tree[rn]=tree[rl]+1;
    if(l==r)return;
    int mid=(l+r)/2;
    if(value>mid)son[0][rn]=son[0][rl],son[1][rn]=++tot,add_tree(son[1][rl],son[1][rn],mid+1,r,value);
    else son[1][rn]=son[1][rl],son[0][rn]=++tot,add_tree(son[0][rl],son[0][rn],l,mid,value);
}
int query(int rl,int rn,int l,int r,int k)
{
    if(l==r)return fp[l];
    int mid=(l+r)/2;
    if(k>tree[son[0][rn]]-tree[son[0][rl]])return query(son[1][rl],son[1][rn],mid+1,r,k-tree[son[0][rn]]+tree[son[0][rl]]);
    else return query(son[0][rl],son[0][rn],l,mid,k);
}
bool cmp(int a,int b)
{
    return line[a]<line[b];
}
int main(void)
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&line[i]);
            p[i]=i;
        }
        sort(p+1,p+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
            vis[p[i]]=i;
            fp[i]=line[p[i]];
        }
        for(int i=1;i<=n;i++)
        {
            rt[i]=++tot;
            add_tree(rt[i-1],rt[i],1,n,vis[i]);
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&L,&R,&K);
            printf("%d\n",query(rt[L-1],rt[R],1,n,K));
        }
    }
    return 0;
}