[POJ2481] cows - 树状数组

---

题目描述

　　农民约翰的奶牛们已经发现，越来越多的草沿山脊(看成是一个数轴)长的特别好。约翰有N头牛(编号从1到N)。每头奶牛都特别喜欢吃一定范围内的草(可能重叠)。这个范围可以看成是一个闭区间[S,E]。
　　例如两头牛cowi和cowj，它们喜欢吃草的范围分别为[Si,Ei]和[Sj,Ej]。如果Si<=Sj，Ej <= Ei且Ei-Si > Ej-Sj，我们就是cowi比cowj强壮。对于每头牛来说，有多少牛是比她强呢？农民约翰需要你的帮助！

---

输入格式

输入文件包含多组测试数据。
　每组测试数据的第一行为一个整数N (1<=N<=10^5),表示奶牛的头数；
　　接下来N行，第i+1行两个整数S和E(0 <= S < E <= 10^5)，表示第i头奶牛的范围，最后用一个0作为文件结束。

---

输出格式

对于每组测试数据输出仅一行为n个用空格分开的整数，第i个数字表示比第i头牛强壮的个数。

---

样例数据

样例输入

3
1 2
0 3
3 4
0

样例输出

1 0 0

---

题目分析

仔细分析题目，会发现第三个条件是废的，只限制了s和e不能分别相等。
故按照s从小到大排序统计比当前e大的个数
或者按照e从大到小排序统计比当前s小的个数
注意树状数组下标不能从0开始，故读入时要+1

---

源代码

#include<algorithm>
#include<iostream>
#include<iomanip>
#include<cstring>
#include<cstdlib>
#include<vector>
#include<cstdio>
#include<cmath>
#include<queue>
#include<map>
using namespace std;
inline const int Get_Int() {
    int num=0,bj=1;
    char x=getchar();
    while(x<'0'||x>'9') {
        if(x=='-')bj=-1;
        x=getchar();
    }
    while(x>='0'&&x<='9') {
        num=num*10+x-'0';
        x=getchar();
    }
    return num*bj;
}
const int maxn=500005;
struct BIT { //树状数组
    int n,c[maxn];
    inline int Lowbit(int x) { //低位操作
        return x&(-x);
    }
    void init(int n) {
        this->n=n;
        memset(c,0,sizeof(c));
    }
    void add(int x,int v) {
        for(int i=x; i<=n; i+=Lowbit(i))c[i]+=v;
    }
    int sum(int x) { //求出1~s的区间和
        int s=0;
        for(int i=x; i; i-=Lowbit(i))s+=c[i];
        return s;
    }
} bit;
int n,Max,ans[500005];
struct Cow { //原题第三个限制条件相当于s和e不能同时等
    int s,e,pos,ans;
} a[500005];
bool cmp1(Cow a,Cow b) {
    return (a.e>b.e)||(a.e==b.e&&a.s<b.s);
}
bool cmp2(Cow s,Cow e) {
    return s.pos<e.pos;
}
int main() {
    while(true) {
        n=Get_Int();
        if(n==0)break;
        for(int i=1; i<=n; i++) {
            a[i].s=Get_Int()+1; //偏移（不能从0开始）
            a[i].e=Get_Int()+1;
            a[i].pos=i;
            Max=max(Max,a[i].e);
        }
        sort(a+1,a+n+1,cmp1);
        bit.init(Max);
        a[1].ans=bit.sum(a[1].s);
        bit.add(a[1].s,1);
        for(int i=2; i<=n; i++) {
            if(a[i].s==a[i-1].s&&a[i].e==a[i-1].e)a[i].ans=a[i-1].ans; //处理相等情况
            else a[i].ans=bit.sum(a[i].s);
            bit.add(a[i].s,1);
        }
        sort(a+1,a+n+1,cmp2);
        for(int i=1; i<=n; i++)printf("%d ",a[i].ans);
        putchar('\n');
    }
    return 0;
}

---