树状数组 or 归并树 —— HDU 4417

• 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4417

• 题意：给出一个数组，每次查询区间 $[l_i，r_i]$ 内大于等于 $h_i$ 的数的个数

• 分析：

• • 解法1：我们可以离线所有查询，然后把查询按照 $h_i$从小到大排序，并且把原来的数组也从小到大排序，从第一个元素开始，判断是否满足当前查询的条件，满足在树状数组中给对应元素的原来位置赋值为1，因为是从小到大，所以如果之前满足的元素之后也肯定满足，这样每次区间求和得到的就是每次查询的结果，然后再按原来的顺序输出即可。复杂度大概为 $O(max(n, m)logn)+O(mlogm)$
• • 解法2：因为是区间查询，我们很容易想到用线段树做，但是每次查询的是不超过指定元素 $h_i$ 的元素个数，$h_i$每次都在变化，所以如果我们每次都判断一遍，相当于是 $O(n^2)$ 的复杂度。那么我们考虑，既然不知道 $h_i$ 的大小，但是我们可以预处理原来数组的顺序，这样每次用二分在区间数组中查找 $h_i$ 的位置即可做到 $O(nlogn)$ 的复杂度。因此，我们在预处理建立线段树的时候 ( 复杂度为 $O(logn)$ )，对于每一个节点对应的区间，可以先做一次排序并且保存下来 ( 每次排序为 $O(nlogn)$ )，这样再加上 $m$ 次查询，我们的最终复杂度为 $O(nlog^2n) + O(mlogn)$

• 树状数组-AC代码：

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include <iostream>
#include <sstream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <bitset>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <list>
#include <ctime>
#include <climits>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 100001
#define MaxM MaxN*10
#define INF 0x3f3f3f3f
#define PI 3.1415926535897932384626
const int mod = 1E9+7;
const double eps = 1e-6;
#define bug cout<<88888888<<endl;
#define debug(x) cout << #x" = " << x << endl;

int n,m; 
int A[MaxN];
struct Node
{
    int id;
    int val;
}node[MaxN];
struct Query
{
    int l,r,h,id,ans;
}query[MaxN];

void init()
{
    CLR(A); 
}
int lowbit(int x)
{
    return x&(-x);
}
void update(int loc, int val)
{
    for(int i=loc;i<=n;i+=lowbit(i))
    {
        A[i]+=val;
    }
}
int sum(int loc)
{
    int ans = 0;
    for(int i=loc;i>0;i-=lowbit(i))
    {
        ans += A[i];
    }
    return ans;
}

int main()
{
    //std::ios::sync_with_stdio(false);
    int T;
    scanf("%d", &T);
    for(int t=1;t<=T;t++)
    {
        scanf("%d%d", &n, &m);
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d", &node[i].val);
            node[i].id = i;
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d", &query[i].l, &query[i].r, &query[i].h);
            query[i].id = i;
        }
        sort(node+1, node+1+n, [](Node a, Node b){return a.val<b.val;});
        sort(query+1, query+1+m, [](Query a, Query b){return a.h<b.h;});
        int loc = 1;
        for(int i=1;i<=m;i++)
        {
            while(loc<=n && query[i].h>=node[loc].val)
            {
                update(node[loc].id, 1);
                loc++;
            }
            query[i].ans = sum(query[i].r+1)-sum(query[i].l);
        }
        sort(query+1, query+1+m, [](Query a, Query b){return a.id<b.id;});
        printf("Case %d:\n", t);
        for(int i=1;i<=m;i++)
        {
            printf("%d\n", query[i].ans);
        }
    }
    //system("pause");
}

• 归并树-AC代码

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include <iostream>
#include <sstream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <bitset>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <list>
#include <ctime>
#include <climits>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 100001
#define MaxM MaxN*10
#define INF 0x3f3f3f3f
#define PI 3.1415926535897932384626
const int mod = 1E9+7;
const double eps = 1e-6;
#define bug cout<<88888888<<endl;
#define debug(x) cout << #x" = " << x << endl;

int N,Q, A[MaxN],a[20][MaxN];
struct Node
{
    int L,R;
    int mid(){return (L+R)/2;}
    int size(){return (R-L)+1;}
}mer_seg[MaxN*4];

void build(int root, int L, int R, int dep)
{
    mer_seg[root].L = L;
    mer_seg[root].R = R;
    if(L!=R)
    {
        int mid = (L+R)>>1;
        build(2*root,L,mid, dep+1);
        build(2*root+1,mid+1,R, dep+1);
        int i = L, j = mid+1, k=L; 
        while(i<=mid && j<=R)
        {
            if(a[dep+1][i]<a[dep+1][j])
                a[dep][k++] = a[dep+1][i++];
            else 
                a[dep][k++] = a[dep+1][j++];
        }
        while(i<=mid) a[dep][k++] = a[dep+1][i++];
        while(j<=R) a[dep][k++] = a[dep+1][j++]; 
    }
    else
    {
        a[dep][L] = A[L];
    }
}
int ans,x;

void query(int root, int s, int e,int dep) 
{
    if(mer_seg[root].L==s && mer_seg[root].R == e) 
    {
        //cout << s << " " << e << endl;
        int cnt = (upper_bound(a[dep]+s, a[dep]+e+1, x) - (a[dep]+s));
        //debug(cnt);
        ans += cnt;
        return ;
    }
    if(e<=mer_seg[root].mid())  query(root*2,s,e,dep+1);
    else if(s>mer_seg[root].mid()) query(root*2+1,s,e,dep+1);
    else
    {
        query(root*2,s,mer_seg[root].mid(),dep+1);
        query(root*2+1,mer_seg[root].mid()+1,e,dep+1);
    }
}

int main()
{
    //std::ios::sync_with_stdio(false);
    int T;
    scanf("%d", &T);
    for(int t=1;t<=T;t++)
    {
        scanf("%d%d", &N, &Q);
        for(int i=1;i<=N;i++) scanf("%d", &A[i]);
        build(1,1,N,1);
        int s,e;
        printf("Case %d:\n", t);
        while(Q--)
        {
            ans = 0;
            scanf("%d%d%d", &s, &e, &x);
            query(1,s+1,e+1,1); 
            printf("%d\n", ans); 
        }
    }
    //system("pause");
}