【NOI2018模拟3.28】Subset

Description

给出三个排列a,b,c，求对于所有的下标集合S，求

$$(max(ai),max(bi),max(ci)),i\subseteq S$$ 的所有可能情况。
n<=1e5

Solution

考虑最大值所属的位置，显然|S|<=3
|S|=1的话就是n
|S|=2就是一个三维偏序
|S|=3比较麻烦，考虑容斥。
合法方案就是三个最大值分别落在三个位置上。
那么不合法方案有两种，一种是三个最大值落在同一个位置上，三维偏序的时候顺便处理就好了。
设这样子的集合为A。
第二种是a,b,c其中两个落在一个位置，第三个落在另一个位置。设这样子的方案为B
这样子我们可以分别统计两维都比a,b，a,c，b,c小的三元组，求和是X
这个东西是二维偏序，直接做就好了。
但是这样子可能会算到第一种。
可以发现第一种会被算3次，也就是3A+B=X
然后就做完了。复杂度O(n log^2 n)

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

int read() {
    char ch;
    for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
    int x=ch-'0';
    for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x;
}

const int N=1e5+5;

struct node{int a,b,c,num;}p[N],q[N];
bool cmp1(node x,node y) {return x.a<y.a;}
bool cmp2(node x,node y) {return x.b<y.b;}

int n,t[N];
ll A,B,X,ans;

void inc(int x) {for(;x<=n;x+=x&-x) t[x]++;}
void dec(int x) {for(;x<=n;x+=x&-x) t[x]--;}
int query(int x) {int res=0;for(;x;x-=x&-x) res+=t[x];return res;}

ll calc(int f1,int f2) {
    if (f1==1) sort(p+1,p+n+1,cmp1);
    if (f1==2) sort(p+1,p+n+1,cmp2);
    ll res=0;
    fo(i,1,n) {
        int now=query((f2==2)?p[i].b:p[i].c);
        res+=(ll)now*(now-1)/2;
        inc((f2==2)?p[i].b:p[i].c);
    }
    fo(i,1,n) dec((f2==2)?p[i].b:p[i].c);
    return res;
}

void solve(int l,int r) {
    if (l==r) return;
    int mid=l+r>>1;
    solve(l,mid);solve(mid+1,r);
    int j=l-1;
    fo(i,mid+1,r) {
        while (j<mid&&p[j+1].b<p[i].b) inc(p[++j].c);
        p[i].num+=query(p[i].c);
    }
    fo(i,l,j) dec(p[i].c);
    int t1=l,t2=mid+1,t3=0;
    while (t1<=mid&&t2<=r) {
        if (p[t1].b<p[t2].b) q[++t3]=p[t1++];
        else q[++t3]=p[t2++];
    }
    while (t1<=mid) q[++t3]=p[t1++];
    while (t2<=r) q[++t3]=p[t2++];
    fo(i,l,r) p[i]=q[i-l+1]; 
}

int main() {
    freopen("subset.in","r",stdin);
    freopen("subset.out","w",stdout);
    n=read();
    fo(i,1,n) p[i].a=read();
    fo(i,1,n) p[i].b=read();
    fo(i,1,n) p[i].c=read();
    sort(p+1,p+n+1,cmp1);

    // part 2 part 3 A
    ans=(ll)n*(n-1)/2;
    solve(1,n);
    fo(i,1,n) ans-=p[i].num,A+=(ll)p[i].num*(p[i].num-1)/2;

    // part 3 X
    X=calc(1,2)+calc(1,3)+calc(2,3);
    B=X-A*3;

    ans+=(ll)n*(n-1)*(n-2)/6-A-B;
    printf("%lld\n",ans+n);
    return 0;
}