小奇遐想 树状数组

题目描述

撷来一缕清风飘渺
方知今日书信未到
窗外三月天霁垂柳新长枝条
风中鸟啼犹带欢笑
——《清风醉梦》
小奇望着青天中的悠悠白云，开始了无限的遐想，在它的视野中，恰好有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

思路：题目要求求一个a<b<d<c的序列，刚开始做的时候以为d只是在c后面且小于c就可以，结果gg，正确思路是先处理一个abxx，xx只要满足大于a大于b即可，因为xx只有两种情况 ，即第一个x大于第2个x 或 第二个x大于第一个x，题目要求的关系就可以转变为abxx-abcd(c<d) 剩下的就是abdc，处理abxx树状数组瞎搞瞎维护，处理abcd时，我们可以枚举c，求小于c的值有多少个，而这个值的含义就是小于这个值的值。。。。举个例子   1...2...7  我们枚举到7，假如小于1的有0个，小于2的有1个，那么getsum(7)就等于0+1，每次维护这个和，就可以得出最后答案。

#include<bits/stdc++.h>
#define read(x) scanf("%d",&x)
using namespace std;
typedef long long LL;
const LL mod=16777216;
const int maxn=1e6+6;
int c[maxn];
int rc[maxn];
const int p=1000000007;

int f[maxn+10];
void init(){//阶乘
    f[0]=1;
    for(int i=1;i<=maxn;i++){
        f[i]=(long long)f[i-1]*i%p;
    }
}

int fast(int a,int n){
    if(n==0)return 1;
    if(n%2)return (long long)a*fast(a,n-1)%p;
    int tmp=fast(a,n/2);
    return (long long)tmp*tmp%p;
}

int C(int n,int m){
    if(n==m||m==0)return 1;
    return ((long long)f[n]*fast(f[m],p-2)%p)*fast(f[n-m],p-2)%p;
}

int lowbit(int x) {
    return x&(-x);
}
void add(int x,int i) {
    while(x<maxn) {
        c[x]+=i;
        x+=lowbit(x);
    }
}
LL getsum(int x) {
    LL sum=0;
    while(x>0) {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}

void add1(int x,int i) {
    while(x<maxn) {
        rc[x]+=i;
        x+=lowbit(x);
    }
}
LL getsum1(int x) {
    LL sum=0;
    while(x>0) {
        sum+=rc[x];
        x-=lowbit(x);
    }
    return sum;
}
int a[maxn];
LL d1[maxn];
LL d2[maxn];
LL num[maxn];
int main() {
    int n;
    init();
    while(read(n) != EOF) {
        LL ans1=0;
        LL ans2=0;
        memset(num,0,sizeof(num));
        memset(c,0,sizeof(c));
        memset(rc,0,sizeof(rc));
        memset(d1,0,sizeof(d1));
        memset(d2,0,sizeof(d2));
        for(int i=1;i<=n;i++) {
            read(a[i]);
            d1[i]+=getsum(a[i]);
            num[i]=num[i-1]+d1[i];
            add(a[i],1);
        }
        memset(c,0,sizeof(c));
        for(int i=n;i>=1;i--) {
            d2[i]+=getsum1(n)-getsum1(a[i]);
            add1(a[i],1);
            //cout<<d1[i]<<" "<<d2[i]<<endl;
        }
        for(int i=1;i<=n;i++) {
            ans1+=d1[i]*(d2[i]*(d2[i]-1)/2);
            ans1%=mod;
        }
        //cout<<ans1<<endl;
        //for(int i=1;i<=n;i++) cout<<d1[i]<<" "<<d2[i]<<endl;
        LL ans=0;
        for(int i=1;i<=n;i++) {
            ans2+=(getsum(a[i]))*d2[i];
            add(a[i],d1[i]);
            ans2%=mod;
        }
        printf("%lld\n",(ans1+mod-ans2)%mod);
    }
}