POJ - 2481 Cows

POJ - 2481 Cows

题意：

有N头牛，每只牛有一个测试值[S,E]，如果对于牛i和牛j来说，它们的测验值满足下面的条件则证明牛i比牛j强壮：Si <= Sj and Ej <= Ei and Ei - Si > Ej - Sj。现在已知每一头牛的测验值，要求输出每头牛有几头牛比它强壮。

思路：

将终点 e 从大到小排序，若终点相同，就让起点s小的在前。以起点建立树状数组（初始全为0），在查询时，只需要求出这个起点所在树状数组的前缀和即为包含这个区间的个数（比这头牛壮的个数），以 s 点建立树状数组。若两区间的起点终点完全相同（实力相当不能计算在内），则（第二个区间）在计数时直接等于前一个区间。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int MAX;
int c[maxn],s[maxn];
struct node
{
    int s,e;
    int pos;
}p[maxn];
bool cmp(node a,node b)
{
    if(a.e==b.e)
        return a.s<b.s;
    else
        return a.e>b.e;
}
int lowbit(int x)
{
    return x&(-x);
}
void add(int x)
{
    while(x<=MAX)
    {
        c[x]++;
        x += lowbit(x);
    }
}
int sum(int x)   //因为按照终点从大到小排序  
{               //所以在x点前的区间的终点都比x的终点大  只需要计算x点前的区间左端点有多少个为1  就有多少个包含以x为起点的区间
    int ans=0;
    while(x>0)
    {
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}
int main()
{
    int n;
    while(~scanf("%d",&n) && n)
    {
        MAX=0;
        memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&p[i].s,&p[i].e);
            MAX=max(MAX,p[i].e);
            p[i].pos=i;
        }
        sort(p+1,p+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
            if(p[i].s==p[i-1].s && p[i].e==p[i-1].e)   //若两区间相同
                s[p[i].pos]=s[p[i-1].pos];
            else  //否则可以求出覆盖本区间的所有牛的个数，由于排序，只能在前面
                s[p[i].pos]=sum(p[i].s+1);     //因为起点从 0 开始所以需要加 1
            add(p[i].s+1);  //将本区间的起点加入树状数组中
        }
        for(int i=1;i<n;i++)
            printf("%d ",s[i]);
        printf("%d\n",s[n]);
    }
    return 0;
}