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);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章