[COGS2580]偏序 II

[COGS2580]偏序 II

题目大意:

\(n(n\le50000)\)个五元组,求五维偏序。

思路:

CDQ分治套CDQ分治套CDQ分治套树状数组。

时间复杂度\(\mathcal O(n\log^4 n)\)

源代码:

#include<cstdio>
#include<cctype>
#include<algorithm>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
typedef long long int64;
const int N=50001;
int n;
int64 ans;
struct Node {
    int a,b,c,d,t1,t2;
};
Node a[N];
inline bool cmp1(const Node &p1,const Node &p2) {
    return p1.a<p2.a;
}
inline bool cmp2(const Node &p1,const Node &p2) {
    return p1.b<p2.b;
}
inline bool cmp3(const Node &p1,const Node &p2) {
    return p1.c<p2.c;
}
class FenwickTree {
    private:
        int val[N];
        int lowbit(const int &x) const {
            return x&-x;
        }
    public:
        void modify(int p,const int &x) {
            for(;p<=n;p+=lowbit(p)) val[p]+=x;
        }
        int query(int p) const {
            int ret=0;
            for(;p;p-=lowbit(p)) ret+=val[p];
            return ret;
        }
};
FenwickTree t;
void cdq3(const int &b,const int &e) {
    if(b==e) return;
    const int mid=(b+e)>>1;
    cdq3(b,mid);
    cdq3(mid+1,e);
    int p=b,q=mid+1;
    for(;q<=e;q++) {
        if(a[q].t1==1||a[q].t2==1) continue;
        for(;p<=mid&&a[p].c<a[q].c;p++) {
            if(a[p].t1==1&&a[p].t2==1) {
                t.modify(a[p].d,1);
            }
        }
        ans+=t.query(a[q].d);
    }
    while(--p>=b) {
        if(a[p].t1==1&&a[p].t2==1) {
            t.modify(a[p].d,-1);
        }
    }
    std::inplace_merge(&a[b],&a[mid]+1,&a[e]+1,cmp3);
}
void cdq2(const int &b,const int &e) {
    if(b==e) return;
    const int mid=(b+e)>>1;
    cdq2(b,mid);
    cdq2(mid+1,e);
    for(register int i=b;i<=mid;i++) a[i].t2=1;
    for(register int i=mid+1;i<=e;i++) a[i].t2=2;
    std::inplace_merge(&a[b],&a[mid]+1,&a[e]+1,cmp2);
    cdq3(b,e);
    std::sort(&a[b],&a[e]+1,cmp2);
}
void cdq1(const int &b,const int &e) {
    if(b==e) return;
    const int mid=(b+e)>>1;
    cdq1(b,mid);
    cdq1(mid+1,e);
    for(register int i=b;i<=mid;i++) a[i].t1=1;
    for(register int i=mid+1;i<=e;i++) a[i].t1=2;
    std::inplace_merge(&a[b],&a[mid]+1,&a[e]+1,cmp1);
    cdq2(b,e);
    std::sort(&a[b],&a[e]+1,cmp1);
}
int main() {
    freopen("partial_order_two.in","r",stdin);
    freopen("partial_order_two.out","w",stdout);
    n=getint();
    for(register int i=1;i<=n;i++) a[i].a=getint();
    for(register int i=1;i<=n;i++) a[i].b=getint();
    for(register int i=1;i<=n;i++) a[i].c=getint();
    for(register int i=1;i<=n;i++) a[i].d=getint();
    cdq1(1,n);
    printf("%lld\n",ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/skylee03/p/9467950.html
ii