luogu-p1533

题目背景
小卡由于公务需要出差,将新家中的狗狗们托付给朋友嘉嘉,但是嘉嘉是一个很懒的人,他才没那么多时间帮小卡喂狗狗。

题目描述
小卡家有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);
    }
}
发布了34 篇原创文章 · 获赞 3 · 访问量 231

猜你喜欢

转载自blog.csdn.net/qq_44641782/article/details/103691890
今日推荐