hdu 4325 Flowers (树状数组+离散化)

题目：http://acm.hdu.edu.cn/showproblem.php?pid=4325

题意：N朵花，M次询问，N行表示第i朵花在这个时间段内开花，M询问在哪个点有几朵花开。

思路：数状数组即可，但需要离散化开花时间，询问也要一起离散化，因为要把查询时间插进开始和结束时间，结束时间往后退。

            离散化：结构体存开始结束时间和对应顺序（id），排序，再1，2，3……标上，作为离散后的下标放入树状数组中。

    比如 1,4,4,8,   1,4,6，离散化为1,2,2,4    1,2,3

            大大节省了空间。

#include<bits/stdc++.h>
using namespace std;
const int maxn=500005;
int n,m,tree[maxn],a[maxn],k;

struct node{
    int x,id;
    bool operator <(const node &a){
        return x<a.x;
    }
}p[maxn];

int lowbit(int i){
    return i&(-i);
}
int update(int i,int x){
    while(i<=k){
        tree[i]+=x;
        i+=lowbit(i);
    }
}
int query(int i){
    int sum=0;
    while(i>0){
        sum+=tree[i];
        i=i-lowbit(i);
    }
    return sum;
}

int main(){
    int t,tt=0;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        int nn=n*2;
        int mm=nn+m;
        for(int i=0;i<mm;i++){   //全存进去
            scanf("%d",&p[i].x);
            p[i].id=i;  //顺便记录下标顺序
        }
        sort(p,p+mm);
        k=1;
        a[p[0].id]=k;
        for(int i=1;i<mm;i++){ 
            if(p[i].x==p[i-1].x)   //原数据相同，离散后当然也要相同
                a[p[i].id]=k;     //这里正好按id顺序存进去
            else
                a[p[i].id]=++k;
        }
        memset(tree,0,sizeof(tree));
        for(int i=0;i<nn-1;i+=2){
            update(a[i],1);
            update(a[i+1]+1,-1);
        }
        printf("Case #%d:\n",++tt);
        for(int i=nn;i<mm;i++)
            printf("%d\n",query(a[i]));
    }
    return 0;
}