牛牛种花 ( 树状数组 + 离散化 )

题目链接
牛牛种花 ( 树状数组 + 离散化 )_第1张图片
解题报告:
考虑对询问离线,将询问和种植的花的位置以x为第一关键词,y为第二关键词升序。
这样保证后面的x不小于前面的x,这样的花只需要统计前面有多少个y小于查询位置的y即可。树状数组维护一下即可。
数据范围较大,注意离散化。

#include
#define LL long long
#define pii pair
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=2e5+5;
const int inf=0x3f3f3f3f;
const LL MOD=1e4+7;

int tree[maxn],n,m,ans[maxn];
vector<int>g;
pii p[maxn];
struct TMD{
    int x,y,id;
}q[maxn];
int lowbit(int x){return x&(-x);}
int getid(int x){return lower_bound(g.begin(),g.end(),x)-g.begin()+1;}
void modify(int x){
    for(int i=x;i<=2e5;i+=lowbit(i)){
        tree[i]++;
    }
}
int get_sum(int x){
    int ans=0;
    for(int i=x;i>=1;i-=lowbit(i)){
        ans+=tree[i];
    }return ans;
}
bool cmp(TMD A,TMD B){
    if(A.x==B.x) return A.y<B.y;
    return A.x<B.x;
}
bool kmp(pii A,pii B){
    if(A.first==B.first) return A.second<B.second;
    return A.first<B.first;
}
int main() {
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&p[i].first,&p[i].second);
        g.push_back(p[i].second);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].x,&q[i].y);q[i].id=i;
        g.push_back(q[i].y);
    }
    sort(p+1,p+n+1,kmp);
    sort(q+1,q+m+1,cmp);
    sort(g.begin(),g.end());
    g.erase(unique(g.begin(),g.end()),g.end());
    int id=1;
    for(int i=1;i<=m;i++){
        while(id<=n&&p[id].first<=q[i].x){
            modify(getid(p[id].second));
            id++;
        }
        ans[q[i].id]=get_sum(q[i].y);
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}

你可能感兴趣的:(树状数组)