首页 > 代码库 > POJ--2481--Cows【线段树】

POJ--2481--Cows【线段树】

链接 :http://poj.org/problem?id=2481

题意:一条直线上长满了三叶草,有n头牛,每头牛都有喜欢的一段三叶草区间 [ si , ei ] ,如果一头牛喜欢的区间包含了另一头牛喜欢的区间,则说明前者比后者强壮,问对于每头牛各有多少头牛比他强壮。


这道题排序之后就是线段树区间查询的裸题,排序:对区间初始位置从小到大排,对区间结束位置从大到小排,然后依次开始遍历,因为现在找的这头牛的起始位置肯定大于等于找过的牛,所以只用看它的结束位置,从它的结束位置查询到线段树末尾n,看比它的结束位置还大的结束位置有几个,就知道几头牛比它强壮,由于只用比较结束位置,更新的时候也更新结束位置就行了。


#include<cstring>
#include<string>
#include<fstream>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cctype>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<stack>
#include<ctime>
#include<cstdlib>
#include<functional>
#include<cmath>
using namespace std;
#define PI acos(-1.0)
#define MAXN 100100
#define eps 1e-7
#define INF 0x7FFFFFFF
#define long long ll;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

struct node{
    int s,e,num;
}a[MAXN];
int ans[MAXN];
int sum[MAXN<<2];
bool cmp(node x,node y){
    if(x.s==y.s)    return x.e>y.e;
    return x.s<y.s;
}
void pushup(int rt){
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt){
    sum[rt] = 0;
    if(l==r)    return ;
    int m = (l+r)>>1;
    build(lson);
    build(rson);
}
void update(int s,int l,int r,int rt){
    if(l==r){
        sum[rt]++;
        return ;
    }
    int m = (l+r)>>1;
    if(s<=m)    update(s,lson);
    else     update(s,rson);
    pushup(rt);
}
int query(int s,int e,int l,int r,int rt){
    if(s<=l&&r<=e){
        return sum[rt];
    }
    int m = (l+r)>>1;
    int cnt = 0;
    if(s<=m)    cnt+=query(s,e,lson);
    if(e>m)     cnt+=query(s,e,rson);
    return cnt;
}
int main(){
    int n,i;
    while(scanf("%d",&n),n){
        for(i=0;i<n;i++){
            scanf("%d%d",&a[i].s,&a[i].e);
            a[i].num = i;
        }
        sort(a,a+n,cmp);
//        for(i=0;i<n;i++){
//            cout<<a[i].s<<" "<<a[i].e<<endl;
//        }
        build(0,n,1);
        for(i=0;i<n;i++){
            if(i&&a[i].s==a[i-1].s&&a[i].e==a[i-1].e)   ans[a[i].num] = ans[a[i-1].num];
            else    ans[a[i].num] = query(a[i].e,n,1,n,1);
            update(a[i].e,1,n,1);
        }
        for(i=0;i<n-1;i++)  printf("%d ",ans[i]);
        printf("%d\n",ans[i]);
    }
    return 0;
}