HDU 4417 Super Mario(离线，线段树/树状数组，思维）

马里奥是世界著名的水管工。
他“魁梧”的身材和令人惊叹的跳跃能力让我们想起了他。
现在可怜的公主又陷入了困境，马里奥需要拯救他的爱人。
我们把通往老板城堡的路看作一条线(长度为n)，在每个整数点i上都有一块砖在高度hi上。
现在的问题是，如果马里奥能跳的最大高度是H，那么[L，R]马里奥可以打多少块砖。
输入。
第一行后面是一个整数T，即测试数据的数量。
对于每个测试数据：？
第一行包含两个整数n，m(1<=n<=10^5，1<=m<=10^5)，n是道路的长度，m是查询的数量。
下一行包含n个整数，每块砖的高度，范围是[0，1000000000]。
接下来m行，每行包含三个整数L，R，H(0<=L<=R<n 0<=H<=1000000000)。
输出量。
对于每种情况，输出“case X：”(X是从1开始的case number)，后面跟着m行，每行包含一个整数。
第i个整数是Mario可以为第i个查询命中的砖数。
1
10 10
0 5 2 7 5 4 3 8 7 7 
2 8 6
3 5 0
1 3 1
1 9 4
0 1 0
3 5 5
5 5 1
4 6 3
1 5 7
5 7 3
Sample Output
Case 1:
4
0
0
3
1
2
0
1
5
1

 

结构体记录数字大小以及位置，排序备用；

初始化树状数组，每个点值都为1，共n个；

将询问以最高值排序，从大到小与数据进行对比，删去小于问题所需的点；

树状数组区间和输出。

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <map>
#include <string>
#include <vector>
#include <iostream>

using namespace std;

struct nnnnn
{
    int num,postition;
}num[200100];

int cmp1(struct nnnnn q,struct nnnnn w)
{
    return q.num>w.num;
}


int ans[200100];
int tree[200100];

struct ques
{
    int l,r,limit,num;
}question[100100];

int cmp(struct ques q,struct ques w)
{
    return q.limit>w.limit;
}


int main()
{
    int i,j,k,n,t,m,l;

    scanf("%d",&t);

    for(l=1;l<=t;l++)
    {
        scanf("%d",&n);scanf("%d",&m);

        for(i=1;i<=n;i++)
        {
            scanf("%d",&num[i].num);
            num[i].postition=i;

            k=i;
            while(k<=n)
            {
                tree[k]+=1;
                k+=(k&(-k));
            }

        }

        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&question[i].l,&question[i].r,&question[i].limit);
            question[i].num=i;
            question[i].l++;
            question[i].r++;
        }

        sort(question+1,question+1+m,cmp);
        sort(num+1,num+1+n,cmp1);

        int st=1;
        for(i=1;i<=m;i++)
        {
             while(num[st].num>question[i].limit)
             {
                k=num[st].postition;
                while(k<=n)
                {
                    tree[k]-=1;
                    k+=(k&(-k));
                }
                st++;
             }


            k=question[i].r;
            while(k>0)
            {
                ans[question[i].num]+=tree[k];
                k-=(k&(-k));
            }

            k=question[i].l-1;
            while(k>0)
            {
                ans[question[i].num]-=tree[k];
                k-=(k&(-k));
            }

        }

        printf("Case %d:\n",l);
        for(i=1;i<=m;i++)
        {
            printf("%d\n",ans[i]);
        }

        memset(tree,0,sizeof tree);
        memset(question,0,sizeof question);
        memset(num,0,sizeof num);
        memset(ans,0,sizeof ans);


    }


    return 0;
}