主席树模板求调,静态区间第k小,WA,感谢qwq
  • 板块学术版
  • 楼主zhanglh
  • 当前回复4
  • 已保存回复4
  • 发布时间2023/8/17 13:41
  • 上次更新2023/11/3 03:10:17
查看原帖
主席树模板求调,静态区间第k小,WA,感谢qwq
726862
zhanglh楼主2023/8/17 13:41
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;

const int N = 100010;

int n, m;
int a[N];
vector<int> num;
int root[N], idx;
struct Node
{
    int l, r;
    int cnt;
}tr[N * 21];

int find(int x)
{
    return lower_bound(num.begin(), num.end(), x) - num.begin();
}

int build(int l, int r)
{
    int q = ++ idx;
    if (l == r) return q;
    int mid = l + r >> 1;
    tr[q].l = build(l, mid), tr[q].r = build(mid + 1, r);
    return q;
}

int insert(int p, int l, int r, int x)
{
    int q = ++ idx;
    tr[q] = tr[p];
    if (l == r)
    {
        tr[q].cnt ++ ;
        return q;
    }
    int mid = l + r >> 1;
    if (x <= mid) tr[q].l = insert(tr[p].l, l, mid, x);
    else tr[q].r = insert(tr[p].r, mid + 1, r, x);
    tr[q].cnt = tr[tr[q].l].cnt + tr[tr[q].r].cnt;
    return q;
}

int query(int q, int p, int l, int r, int k)
{
    if (l == r) return r;
    int cnt = tr[tr[q].l].cnt - tr[tr[p].l].cnt;
    int mid = l + r >> 1;
    if (k <= cnt) return query(tr[q].l, tr[p].l, l, mid, k);
    else return query(tr[q].r, tr[p].r, mid + 1, r, k);
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
        num.push_back(a[i]);
    }
    
    sort(num.begin(), num.end());
    num.erase(unique(num.begin(), num.end()), num.end());
    
    root[0] = build(0, num.size() - 1);
    
    for (int i = 1; i <= n; i ++ )
        root[i] = insert(root[i - 1], 0, num.size() - 1, find(a[i]));
        
    while (m -- )
    {
        int l, r, k;
        scanf("%d%d%d", &l, &r, &k);
        printf("%d\n", num[query(root[r], root[l - 1], 0, num.size() - 1, k)]);
    }
    
    return 0;
}
2023/8/17 13:41
加载中...