[POJ2481] cows - 树状数组

来源:互联网 发布:轰炸机软件 编辑:程序博客网 时间:2024/06/05 19:16

题目描述

  农民约翰的奶牛们已经发现,越来越多的草沿山脊(看成是一个数轴)长的特别好。约翰有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;}

0 0