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

点击打开链接

题意: 有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;
}