小奇遐想（树状数组+容斥原理）

                                                         小奇遐想

                                                         时间限制: 1 Sec  内存限制: 128 MB
                                                                          提交: 239  解决: 42
                                                         [提交] [状态] [讨论版] [命题人:admin]

题目描述

撷来一缕清风飘渺
方知今日书信未到
窗外三月天霁垂柳新长枝条
风中鸟啼犹带欢笑
——《清风醉梦》
小奇望着青天中的悠悠白云，开始了无限的遐想，在它的视野中，恰好有n朵高度不同的白云排成一排，他想从左到右选出四朵白云a，b，c，d，使得h_a<h_b<h_d<h_c，即看起来像是彩虹的形状！它想知道有多少种方案数。

 

 

输入

第一行包括1个整数n。
第二行包括n个整数，第i个正数表示h_i，保证这n个整数是n的一个全排列。

输出

输出一个整数表示答案。（mod 16777216）

样例输入

5
1 5 3 2 4

样例输出

0

提示

对于10%的数据n<=600；对于40%的数据n<=5000；
对于100%的数据n<=200000。

                                                                                     [提交]    [状态]

在一序列中求满足1243大小关系的串

利用容斥的思想，转换为先求12xx（包括1234，1243），然后再减去1234

先利用一次树状数组求出每个数前面比他小的数字，存在l[i]数组里

通过计算得到每个数后面比他大的数字，存在r[i]数组里

12xx很好计算，就是取数相乘即可

然后计算1234，因为每个数字比他小的已经存在l[i]数组里，所以一种想法是用树状数组求一次比他大的，然后在求一次

优化之后的是利用l[i]数组计算每个数前面的不下降子串的个数，然后再乘后面比他大的数字个数

最后两者相减

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxx=2e5+100;
const int INF=1e9;
const int MOD=16777216;
int n;
int a[maxx];
int l[maxx],r[maxx];
int sum[maxx];
int lowbit(int x)
{

    return x&(-x);

}

void add(int x,int i)
{
    while(x<maxx)
    {
        sum[x]+=i;
        x+=lowbit(x);
    }
}

ll query(int x)
{

    ll ret=0;
    while(x>0)
    {
        ret+=sum[x];
        x-=lowbit(x);
    }

    return ret;

}


int main()
{

    scanf("%d",&n);
    for(int i=1; i<=n; i++){

        scanf("%d",&a[i]);
        l[i]=query(a[i]);//比 a[i]小的个数
        r[i]=n-i-(a[i]-1-l[i]);// a[i]-1 序列中一共有几个比他小的 a[i]-1-l[i] 右边比他小的个数 n-i i后面有多少个数
        add(a[i],1);

    }

    ll ans1=0;

    for(int i=1; i<=n; i++){//枚举第二位 i表示第二位是啥

        ans1=(ans1+l[i]*r[i]*(r[i]-1)/2)%MOD;

    }

    memset(sum,0,sizeof(sum));

    ll ans2=0;

    for(int i=1; i<=n; i++){//枚举第三位 i 表示第三位是啥

        ans2=(ans2+query(a[i])*r[i])%MOD;  // 先询问在建树所求的区间和不包括 放入的本身 反之则包括
        add(a[i],l[i]);

    }

    printf("%lld\n",(ans1-ans2+MOD)%MOD);

    return 0;

}