[ツリー配列]HOJ 10069星の等級

2519 ワード

転送ゲート:http://acm.hnu.cn/online/?action=problem&type=show&id=10069&courseid=0
いくつかの二次元平面上の点を与えて、a.x<=b.x&a.y>=b.yならば、aの等級はbより高いと言います.
問題の考え方:星をyの順とxの順に並べ替えて、x軸に対して木のような配列を立てて、順番に星ごとに木のような配列に挿入します.(挿入する前に1-現在の座標と区間とを統計して記録してください.)そして同じ星の記録データを彼らの中の最初の木の配列に挿入された星の値に変えてください.離散化が必要です.すなわち前述の樹状配列は離散化後のx軸の下に構築されます.
卵の痛みの種類:
1.比較関数に最後のIDx<s.idxを付けないとWA(こう書くと並べ替えの結果に問題があるはずです)
2.最初は見ていませんでした.
最初の問題は卵が痛いです.これから注意してください.
コード:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 111111;
struct Star{
    int x,y,idx;
    bool operator < (const Star &s) const{
        if(y!=s.y){
            return y>s.y;
        }else if(x!=s.x){
            return x<s.x;
        }else{
            return idx<s.idx;
        }// 。。。。。
    }
    bool operator == (const Star &s) const{
        return x==s.x&&y==s.y;
    }
}stars[MAXN];
int arr[MAXN],c[MAXN],res[MAXN],N,M;
int bs(int x){
    int l = 0, r = M-1,m;
    while(l<=r){
        m = (l+r)>>1;
        if(arr[m]==x)return m;
        else if(x<arr[m])r = m-1;
        else l = m+1;
    }
    return -1;
}
int lowbit(int x){
    return x&(-x);
}
int sum(int pos){
    int s = 0;
    while(pos>0){
        s+=c[pos];
        pos-=lowbit(pos);
    }
    return s;
}
void add(int pos){
    while(pos<=M){
        c[pos]++;
        pos+=lowbit(pos);
    }
}
int main(){
    while(scanf("%d",&N)!=EOF){
        memset(c,0,sizeof(c));
        M = 0;
        for(int i=0;i<N;i++){
            scanf("%d%d",&stars[i].x,&stars[i].y);
            stars[i].idx = i;
            arr[M++] = stars[i].x;
        }
        sort(arr,arr+M);
        int temp = 1;
        for(int i=1;i<M;i++)if(arr[i]!=arr[i-1])arr[temp++] = arr[i];
        M = temp;
        sort(stars,stars+N);
        for(int i=0;i<N;i++){
            int pos = bs(stars[i].x)+1;
            res[stars[i].idx] = sum(pos);
            add(pos);
        }
        for(int i=1;i<N;i++){
            if(stars[i]==stars[i-1])res[i]=res[i-1];
        }
        for(int i=0;i<N;i++){
            if(i) printf(" %d",res[i]);
            else printf("%d",res[i]);
        }
        printf("
"); } return 0; }