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;
}

POJ--2481--Cows【线段树】,布布扣,bubuko.com

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

标签：线段树   单点更新   区间查询   

原文地址：http://blog.csdn.net/zzzz40/article/details/38129213