hdu_4417 Super Mario (树状数组)

http://acm.hdu.edu.cn/showproblem.php?pid=4417

 

题意：
给一个数列a[1..n]，再给一些查询,每次查询包含三个元素l,r,h;表示求[l,r]区间内小于h数的个数。
思路：
用数状数组记录当前比h小的数的个数。以前做过一道用数状数组求逆序数的题目，这题可以类比思维的想下，将砖块高度排序，再将要
查询的高度排序，再将砖块下标加入数状数组，当砖块高度达到查询高度时，查询下并保存结果。

我的代码

 

/*
program:hdu_4417
author:BlackAndWhite
*/
#include<stdio.h>
#include<algorithm>
#include<string.h>
#define lowbit(x) (x)&(-x)
#define MAXN 500001
using namespace std;
int c[MAXN],ans[MAXN];
struct node
{
    int id,tmp,h,l,r;
    //id原顺序，tmp原始数据，h离散后
}a[2*MAXN],b[MAXN];
int i,j,n,m,t,nn,js=1;
void Add(int i,int val,int n)
{
    for(;i<=n;i+=lowbit(i)) c[i]+=val;
}
int Sum(int i)
{
    int ret=0;
    for(;i>0;i-=lowbit(i)) ret+=c[i];
    return ret;
}
int cmp(node a,node b)
{
    if(a.tmp==b.tmp) return a.id<b.id;
    return a.tmp<b.tmp;
}
int cmp1(node a,node b)
{
    return a.id<b.id;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<n;i++)
        {
            scanf("%d",&a[i].tmp);
            a[i].id=i;
        }
        for(;i<n+m;i++)
        {
            scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].tmp);
            a[i].id=i;
        }
        sort(a,a+m+n,cmp);
        j=1;a[0].h=1;
        for(i=1;i<n+m;i++)
            if(a[i].tmp==a[i-1].tmp) a[i].h=j;
            else a[i].h=++j;
        sort(a,a+m+n,cmp1);
        sort(a+n,a+n+m,cmp);
        sort(a,a+n,cmp);
        for(i=0;i<n;i++)
            b[i]=a[i];//离散后的砖块高度
        b[i].h=1000000001;
        nn=MAXN-1;
        memset(c,0,sizeof(c));
        memset(ans,0,sizeof(ans));
        j=0;
        for(i=n;i<n+m;i++)
        {
            while(b[j].h<=a[i].h)
                Add(b[j++].id+1,1,nn);
            ans[a[i].id-n]=Sum(a[i].r+1)-Sum(a[i].l);
        }
        printf("Case %d:\n",js++);
        for(i=0;i<m;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}