HDU5032 Always Cook Mushroom(树状数组&&离线)

树状数组＋询问离线。一个优化是需要的，就是先对1000*1000个点先排序，而不是每次都生成这1000*1000个点然后和询问一起排序，那样会tle.

#include <iostream>

#include <cstring>

#include <string>

#include <vector>

#include <cstdio>

#include <algorithm>

#include <cmath>

using namespace std;



#define maxn 1200

#define ll long long



ll A,B;



struct Node

{

    ll a,b;

    int t;

    ll v;

    int id;

    bool operator < (const Node& y) const{

        if(b*y.a!=a*y.b) return b*y.a<a*y.b;

        else return t<y.t;

    }

}x[120000],y[1005000];



int tot,tot2;



ll ans[200000];



ll bit[maxn];



void add(int i,ll v){

    while(i<=1000){

        bit[i]+=v;

        i+=i&(-i);

    }

}



ll sum(int i){

    ll ret=0;

    while(i>0){

        ret+=bit[i];

        i-=i&(-i);

    }

    return ret;

}



int main()

{

    int T;cin>>T;int ca=0;

    tot2=0;

    for(int i=1;i<=1000;++i){

        for(int j=1;j<=1000;++j){

            y[tot2].a=i;y[tot2].b=j;

            y[tot2++].t=1;

        }

    }

    sort(y,y+tot2);

    while(T--){

        scanf("%I64d%I64d",&A,&B);

        tot=0;

        int m;scanf("%d",&m);

        int ai,bi,xi;

        for(int i=0;i<m;++i){

            scanf("%d%d%d",&ai,&bi,&xi);

            x[tot].a=ai;

            x[tot].b=bi;

            x[tot].id=i;

            x[tot].t=2;

            x[tot++].v=xi;

        }



        sort(x,x+tot);

        memset(bit,0,sizeof(bit));

        int id=0;

        for(int i=0;i<tot;++i){

            while(id<tot2&&y[id]<x[i]){

                add(y[id].a,(y[id].a+A)*(y[id].b+B));

                ++id;

            }

            ans[x[i].id]=sum(x[i].v);

        }

        printf("Case #%d:\n",++ca);

        for(int i=0;i<m;++i){

            printf("%I64d\n",ans[i]);

        }

    }

    return 0;

}