阿生在HBU教授数据结构课程。他给可怜的孩子们留了一道毒瘤作业,要求同学给他编写一个新的数据结构,该结构能够在数组的某个区间内快速返回k阶统计量。
也就是说,给定一个由不同整数组成的数组a[1⋯n],给出一系列问题Q(i,j,k)需要给出区间[i,j]中的第k小的数字。
例如,考虑数组a=(1,5,2,6,3,7,4)。设问题为Q(2,5,3)。区间a[2⋯5]是(5,2,6,3)。问题的答案是5。
输入格式:
输入文件的第一行包含n数组的大小,m要回答的问题数(1<=n<=100000,1<=m<=5000)。
第二行包含n个不同的整数,其绝对值不超过109.
以下m行包含问题描述,每个描述由三个数字组成:i、j和k(1<=i<=j<=n,1<=k<=j−i+1),并表示问题Q(i、j、k)。
输出格式:
对于每个问题,输出在区间[i,j]段中的第k小数字
输入样例:
在这里给出一组输入。例如:
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
输出样例:
在这里给出相应的输出。例如:
5
6
3
思路:区间第k小,显然是主席树,板子
#include<bits/stdc++.h>
using namespace std;
#define x first
#define y second
#define endl '\n'
#define rep(i,a,n) for (int i = a; i < n; i ++ )
#define repn(i,a,n) for (int i = a; i <= n; i ++ )
#define pb push_back
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long ll;
#define all(x) (x).begin(),(x).end()
typedef pair<int,int> PII;
ll gcd(ll a,ll b) { return b ? gcd(b,a % b) : a; }
const int mod = 1e9+7;
const int N = 100010, M = 10010;
int n, m;
int a[N];
vector<int> nums;
struct Node
{
int l, r;
int cnt;
}tr[N * 4 + N * 17];
int root[N], idx;
int find(int x)
{
return lower_bound(nums.begin(), nums.end(), x) - nums.begin();
}
int build(int l, int r)
{
int p = ++ idx;
if (l == r) return p;
int mid = l + r >> 1;
tr[p].l = build(l, mid), tr[p].r = build(mid + 1, r);
return p;
}
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 - cnt);
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ )
{
scanf("%d", &a[i]);
nums.push_back(a[i]);
}
sort(nums.begin(), nums.end());
nums.erase(unique(nums.begin(), nums.end()), nums.end());
root[0] = build(0, nums.size() - 1);
for (int i = 1; i <= n; i ++ )
root[i] = insert(root[i - 1], 0, nums.size() - 1, find(a[i]));
while (m -- )
{
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
printf("%d\n", nums[query(root[r], root[l - 1], 0, nums.size() - 1, k)]);
}
return 0;
}