Educational Codeforces Round 10D 离散化+树状数组

太过爱你忘了你带给我的痛 2022-07-28 01:05 154阅读 0赞

[点击打开链接][Link 1]

[  
][Link 1]

题意: 有n条线段, n<=200000, 每条线段有左端点l 和右端点r, -1e9 <= l<r<1e9, 并且r各不相同, 求每条线段所覆盖的其他线段的数量.

分析: 求每条线段覆盖的其他线段的数量, 就是求其他线段左端点 l'>=l 并且右端点 r'<r的数量, 如果吧左右端点对应于平面上的点, 那就是求在l,r该点右下方的点的数目,

我们可以先把右端点离散化, 按照 l 从小到大排序之后倒着添加离散化之后的右端点到树状数组中, 然后每一个端点要统计小于右端点的数目.

#include<bitset>
    #include<map>
    #include<vector>
    #include<cstdio>
    #include<iostream>
    #include<cstring>
    #include<string>
    #include<algorithm>
    #include<cmath>
    #include<stack>
    #include<queue>
    #include<set>
    #define inf 0x3f3f3f3f
    #define mem(a,x) memset(a,x,sizeof(a))
    #define F first
    #define S second
    using namespace std;
    
    typedef long long ll;
    typedef unsigned long long ull;
    typedef pair<int,int> pii;
    
    inline int in()
    {
        int res=0;char c;int f=1;
        while((c=getchar())<'0' || c>'9')if(c=='-')f=-1;
        while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
        return res*f;
    }
    const int N=200010,MOD=1e9+7;
    
    struct seg
    {
        int l,r,id;
    }a[N];
    int ans[N];
    
    bool cmp(seg a,seg b)
    {
        if(a.l!=b.l) return a.l<b.l;
        return a.r<b.r;
    }
    int r[N];
    
    int bit[N];
    void add(int i)
    {
        for(;i<N;i+=i&-i) bit[i]++;
    }
    int sum(int i)
    {
        int s=0;
        for(;i;i-=i&-i) s+=bit[i];
        return s;
    }
    
    int main()
    {
        int n=in(),cnt=0;
        for(int i=0;i<n;i++)
        {
            a[i].l=in();
            a[i].r=in();
            a[i].id=i;
            r[cnt++]=a[i].r;
        }
        sort(r,r+cnt);
        int m=unique(r,r+cnt)-r;
        sort(a,a+n,cmp);
    
        for(int i=n-1;i>=0;i--)
        {
            int pos=lower_bound(r,r+m,a[i].r)-r;
            pos++;
            ans[a[i].id]=sum(pos);
            add(pos);
        }
        for(int i=0;i<n;i++)
        {
            printf("%d\n",ans[i]);
        }
        return 0;
    }

[Link 1]: http://codeforces.com/contest/652/problem/D