JZOJ5608. 【NOI2018模拟3.28】Subset

题目

Data Constraint

对于 10% 的数据满足 n ≤ 20
对于 30% 的数据满足 n ≤ 2000
另有 20% 的数据满足 a i = b i
对于 100% 的数据满足 n ≤ 100000

题解

可以知道S最大只有3，
即a,b,c的最大值分别占一列，
再多一列，上面没有最大值，那么对答案也没有影响。

分类讨论
1、S的大小为1，很显然有n种，
2、S的大小为2，有三种情况ab,c / ac,b / bc,a
以ab,c这种情况为例：
按照a从小到大排序，
枚举i，i列必选，那么i这一列中存在a，b的最大值，c的最大值位于另外一列。
现在要找的就是在i前面，比b小，比c大的数量，用树套树解决。
3、S的大小为3，这里的情况比较多，所以就反过来求，
设A+B为不满足三个最大值分别在三列的数量，
A为三个最大值同在一列的数量，
X为ab,?,? / bc,?,? / ac,?,? 的总数量，
通过容斥可以知道B+3A=X
而且A,X是很容易求，那么B也就可以知道，
最后就可以求出方案数。

code

#include <queue>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <time.h>
#define ll long long
#define N 100003
#define M 103
#define db double
#define P putchar
#define G getchar
#define inf 998244353
#define pi 3.1415926535897932384626433832795
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

struct node
{
    int l,r,s;
}t[N*200];

struct arr
{
    int a,b,c;
}a[N];

ll ans,s,A,B,X;
int n,opx,opl,opr,ops,tot,root[N];
bool p;

int x_(int x){return x&(-x);}

bool cmp(arr x,arr y)
{
    return x.a<y.a;
}

ll C(ll x){return ((x-1)*x)>>1;}

void work(int x,int l,int r)
{
    if(t[x].s==0)return;
    if(opl<=l && r<=opr)
    {
        ops+=t[x].s;
        return;
    }
    int m=(l+r)>>1;
    if(opl<=m)work(t[x].l,l,m);
    if(m<opr)work(t[x].r,m+1,r);
}

void work_(int x,int l,int r)
{
    if(l==r)
    {
        t[x].s++;
        return;
    }
    int m=(l+r)>>1;
    if(opl<=m)
    {
        if(!t[x].l)t[x].l=++tot;
        work_(t[x].l,l,m);
    }
    else
    {
        if(!t[x].r)t[x].r=++tot;
        work_(t[x].r,m+1,r);
    }
    t[x].s++;
}

ll get(int x,int l,int r)
{
    opl=l;opr=r;opx=2;ops=0;
    for(int i=x;i;i=i-x_(i))
        work(root[i],1,n);
    return ops;
}

void ins(int x,int y,int z)
{
    opl=opr=y;
    opx=1;ops=z;
    for(int i=x;i<=n;i=i+x_(i))
        work_(root[i],1,n);
}

void work_2()
{
    memset(t,0,sizeof(t)),tot=n;
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        s+=get(a[i].b,a[i].c+1,n);
        X+=C(get(a[i].b,1,n));
        ins(a[i].b,a[i].c,1);
    }
}

ll work_3()
{
    ll ss=0;
    memset(t,0,sizeof(t)),tot=n;
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        ss+=C(get(a[i].b,1,a[i].c-1));
        s+=get(a[i].b,a[i].c+1,n);
        X+=C(get(a[i].b,1,n));
        ins(a[i].b,a[i].c,1);
    }
    return ss;
}

int main()
{
    freopen("subset.in","r",stdin);
    freopen("subset.out","w",stdout);

    read(n);
    for(int i=1;i<=n;i++)root[i]=i;tot=n;
    for(int i=1;i<=n;i++)read(a[i].a);
    for(int i=1;i<=n;i++)read(a[i].b);
    for(int i=1;i<=n;i++)read(a[i].c);

    s=X=0;
    work_2();
    for(int i=1;i<=n;i++)swap(a[i].b,a[i].c);
    work_2();
    for(int i=1;i<=n;i++)swap(a[i].a,a[i].c);
    A=work_3();

    B=X-3*A;

    ans=(ll)(n-1)*(n-2)*n/2/3-A-B+n+s;

    write(ans);

    return 0;
}