Super Mario （HDU - 4417，主席树查询区间小于 k 的个数）

最新推荐文章于 2019-10-16 20:57:06 发布

原创最新推荐文章于 2019-10-16 20:57:06 发布 · 142 阅读

0 ·

CC 4.0 BY-SA版权

主席树专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用离散化和线段树进行区间查询优化的方法，特别适用于查询区间内元素个数小于特定阈值的问题。通过离散化减少数据范围，结合线段树的高效查询与更新能力，实现了对大规模数据集的有效处理。

一.题目链接：

HDU-4417

二.题目大意：

查询区间小于 k 的个数

三.分析：

这里只是存个板子~~

四.代码实现：

#include <set>
#include <map>
#include <ctime>
#include <queue>
#include <cmath>
#include <stack>
#include <bitset>
#include <vector>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define eps 1e-8
#define lc k * 2
#define rc k * 2 + 1
#define pi acos(-1.0)
#define ll long long
#define ull unsigned long long
using namespace std;

const int M = (int)1e5;
const int mod = 99991;
const int inf = 0x3f3f3f3f;

int a[M + 5];
int b[M + 5];
int n, m, len;

int cnt;
struct node
{
    int sum;
    int l, r;
}tree[M * 40 + 5];
int root[M + 5];

void discrete()
{
    memcpy(b, a, sizeof(a));
    sort(b + 1, b + n + 1);
    len = unique(b + 1, b + n + 1) - (b + 1);
}

int tofind(int x)
{
    return upper_bound(b + 1, b + len + 1, x) - b;
}

void update(int &x, int y, int l, int r, int v)
{
    tree[++cnt] = tree[y];
    tree[cnt].sum++;
    x = cnt;
    if(l == r)
        return;
    int mid = (l + r) >> 1;
    if(v <= mid)
        update(tree[x].l, tree[y].l, l, mid, v);
    else
        update(tree[x].r, tree[y].r, mid + 1, r, v);
}

int query(int x, int y, int l, int r, int a, int b)
{
    if(l >= a && r <= b)
        return tree[y].sum - tree[x].sum;
    int sum = 0;
    int mid = (l + r) >> 1;
    if(a <= mid)
        sum += query(tree[x].l, tree[y].l, l, mid, a, b);
    if(mid < b)
        sum += query(tree[x].r, tree[y].r, mid + 1, r, a, b);
    return sum;
}

int main()
{
    int T;
    scanf("%d", &T);
    for(int ca = 1; ca <= T; ++ca)
    {
        cnt = 0;
        scanf("%d %d", &n, &m);
        for(int i = 1; i <= n; ++i)
            scanf("%d", &a[i]);
        discrete();
        for(int i = 1; i <= n; ++i)
            update(root[i], root[i - 1], 1, n, tofind(a[i]) - 1);
        int l, r, k;
        printf("Case %d:\n", ca);
        while((m--) > 0)
        {
            scanf("%d %d %d", &l, &r, &k);
            k = tofind(k) - 1;
            l++, r++;
            if(k)
                printf("%d\n", query(root[l - 1], root[r], 1, n, 1, k));
            else
                printf("0\n");
        }
    }
    return 0;
}