题目背景
小卡由于公务需要出差,将新家中的狗狗们托付给朋友嘉嘉,但是嘉嘉是一个很懒的人,他才没那么多时间帮小卡喂狗狗。
题目描述
小卡家有N只狗,由于品种、年龄不同,每一只狗都有一个不同的漂亮值。漂亮值与漂亮的程度成反比(漂亮值越低越漂亮),吃饭时,狗狗们会按顺序站成一排等着主人给食物。
可是嘉嘉真的很懒,他才不肯喂这么多狗呢,这多浪费时间啊,于是他每次就只给第i只到第j只狗中第k漂亮的狗狗喂食(好狠心的人啊)。而且为了保证某一只狗狗不会被喂太多次,他喂的每个区间(i,j)不互相包含。
输入格式
第一行输入两个数n,m,你可以假设n<300001 并且 m<50001;m表示他喂了m次。
第二行n个整数,表示第i只狗的漂亮值为ai。
接下来m行,每行3个整数i,j,k表示这次喂食喂第i到第j只狗中第k漂亮的狗的漂亮值。
输出格式
M行,每行一个整数,表示每一次喂的那只狗漂亮值为多少。
输入
7 2
1 5 2 6 3 7 4
1 5 3
2 7 1
输出
3
2
这道题是一道主席树的模板题,直接套主席树的板子就可以了,最开始我把 l 打成了1,一直WA,WA到我怀疑人生。
主席树代码:
#pragma GCC optimize(2)
#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e5 + 7;
std::mt19937 rnd(233);
struct tree
{
int l, r, date;
} t[maxn << 5];
int a[maxn], b[maxn], T[maxn], cnt;
inline void build(int &root,int l,int r)
{
root = ++cnt;
t[root].date = 0;
if(l!=r){
int mid = l + r >> 1;
build(t[root].l, l, mid);
build(t[root].r, mid + 1, r);
}
}
inline void update(int pre,int &now,int l,int r,int k)
{
now = ++cnt;
t[now] = t[pre];
t[now].date++;
if(l!=r){
int mid = l + r >> 1;
if(k<=mid){
update(t[pre].l, t[now].l, l, mid, k);
}
else{
update(t[pre].r, t[now].r, mid + 1, r, k);
}
}
}
inline void query(int pre,int now,int l,int r,int k)
{
if(l==r){
printf("%d\n", b[l]);
return;
}
int pr = t[pre].l, nr = t[now].l;
int sum = t[nr].date - t[pr].date;
int mid = l + r >> 1;
if(sum>=k){
query(pr, nr, l, mid, k);
}
else{
query(t[pre].r, t[now].r, mid + 1, r, k - sum);
}
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n;i++){
scanf("%d", a + i);
b[i] = a[i];
}
sort(b + 1, b + 1 + n);
int len = unique(b + 1, b + 1 + n) - b - 1;
build(T[0], 1, len);
//cout << len << endl;
for (int i = 1; i <= n;i++){
int k = lower_bound(b + 1, b + 1 + len, a[i]) - b;
// cout << k << endl;
update(T[i - 1], T[i], 1, len, k);
}
while(m--){
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
query(T[l - 1], T[r], 1, len, k);
}
}