BZOJ3262陌上花開

三維偏序的模板。ios

固然各類樹套樹均可以搞,這裏用CDQ分治弄一下。數組

首先利用排序使第一維有序,而後利用cdq函數開始執行相似歸併排序的操做,因爲左區間的第一維必定小於右區間的第一維,因此咱們在歸併過程當中直接比較第二維,而後利用樹狀數組維護第三維,使之知足三維偏序,直接統計答案,執行一次cdq操做便可。ide

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<set>
#include<map>
#define lowbit(x) (x&(-x))
using namespace std;
struct Flower{
    int a,b,c,w,ans;
    friend bool operator < (const Flower &x,const Flower &y){
        if(x.a==y.a){
            if(x.b==y.b) return x.c<y.c;
            else return x.b<y.b;
        }else return x.a<y.a;
    }
}q[100500],tmp[100500];
int read(){
    int sum=0,f=1;char x=getchar();
    while(x<'0'||x>'9'){
        if(x=='-') f=-1;
        x=getchar();
    }while(x>='0'&&x<='9'){
        sum=(sum<<3)+(sum<<1)+x-'0';
        x=getchar();
    }return sum*f;
}
int n,lin,wmz,N,level[100500],t[205000];
bool gv[100500];
void add(int pos,int val){
    for(int i=pos;i<=wmz;i+=lowbit(i))
        t[i]+=val;
}
int ask(int pos){
    int ans=0;
    for(int i=pos;i;i-=lowbit(i))
        ans+=t[i];
    return ans;
}
void cdq(int l,int r){
    if(l==r) return ;
    int mid=l+r>>1;
    int i=l,j=mid+1,tot=l;
    cdq(l,mid);cdq(mid+1,r);
    while(i<=mid&&j<=r){
        if(q[i].b<=q[j].b) tmp[tot++]=q[i++];
        else tmp[tot]=q[j++],gv[tot++]=1;
    }
    while(i<=mid) tmp[tot++]=q[i++];
    while(j<=r) tmp[tot]=q[j++],gv[tot++]=1;
    /*cout<<endl;
    for(int i=l;i<=r;i++)
        cout<<tmp[i].a<<" "<<tmp[i].b<<" "<<tmp[i].c<<endl;*/
    for(int i=l;i<=r;i++){
        if(gv[i]) tmp[i].ans+=ask(tmp[i].c);
        else add(tmp[i].c,tmp[i].w);
    }
    for(int i=l;i<=r;i++){
        if(!gv[i]) add(tmp[i].c,-tmp[i].w);
        else gv[i]=0;
        q[i]=tmp[i];
    }
}
int main(){
    n=read();
    wmz=read();
    for(int i=1;i<=n;i++)
        q[i].a=read(),q[i].b=read(),q[i].c=read(),q[i].w=1;
    sort(q+1,q+1+n);
/*    cout<<endl;
    for(int i=1;i<=n;i++)
        cout<<q[i].a<<" "<<q[i].b<<" "<<q[i].c<<endl;*/
    lin=1;
    for(int i=2;i<=n;i++)
        if(q[i].a==q[lin].a&&q[i].b==q[lin].b&&q[i].c==q[lin].c) q[lin].w++;
        else q[++lin]=q[i];
    N=n;
    n=lin;
/*    cout<<endl;
    for(int i=1;i<=n;i++)
        cout<<q[i].a<<" "<<q[i].b<<" "<<q[i].c<<" "<<q[i].w<<endl;*/
    cdq(1,n);
    for(int i=1;i<=n;i++)
        level[q[i].ans+q[i].w-1]+=q[i].w;
    for(int i=0;i<N;i++)
        printf("%d\n",level[i]);
    return 0;
}
View Code
相關文章
相關標籤/搜索