hdu 5792 World is Exploding 树状数组

最新推荐文章于 2019-06-29 12:27:16 发布

varinic

最新推荐文章于 2019-06-29 12:27:16 发布

阅读量584

点赞数

分类专栏：树状数组文章标签：树状数组

本文链接：https://blog.csdn.net/Mtrix/article/details/52096560

版权

树状数组专栏收录该内容

6 篇文章 0 订阅

订阅专栏

计算像a，b这样上升的有la对，像c，d这样下降的有lb对，ans=la*lb。这样是有重复的，重复的就是a与c重合，a与d重合，b与c重合，b与d重合这四种情况。那么减去这四种情况就ok了。可以用树状数组预处理出每一位i的左边比a[i]大的有多少La[]，少的有多少Li[]，右边比a[i]大的有多少Ra[]，少的有多少Li[]。最后ans减去Ri[i]*Ra[i]; Li[i]*La[i]; Li[i]*Ri[i]; La[i]*Ra[i] 就行了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<string>
#include<algorithm>
#include<set>
#include<map>
#include<cstring>
#include<queue>
#include<stack>
#include<list>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

typedef long long ll;
const int maxn=50000+1000;
const ll inf=1e9;
const ll mod=1e9+7;

int a[maxn];
int b[maxn];
ll Li[maxn],La[maxn],Ri[maxn],Ra[maxn];
int cnt[maxn];
int bit[maxn];

int sum(int i){
    int s=0;
    while(i>0){
        s+=bit[i];
        i-=i&-i;
    }
    return s;
}

void add(int i,int x){
    while(i<maxn){
        bit[i]+=x;
        i+=i&-i;
    }
}

int main()
{
    int n;
    while(~scanf("%d",&n)){
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b,b+n);
        int m=unique(b,b+n)-b;
        for(int i=0;i<n;i++){
            a[i]=lower_bound(b,b+m,a[i])-b;
            a[i]+=1;
            //cout<<a[i]<<endl;
        }
        ll la=0,lb=0;
        memset(cnt,0,sizeof(cnt));
        memset(bit,0,sizeof(bit));
        for(int i=0;i<n;i++){
            Li[i]=sum(a[i]-1);
            La[i]=i-Li[i]-cnt[a[i]];
            add(a[i],1);
            cnt[a[i]]++;
            la+=Li[i];
        }
        memset(cnt,0,sizeof(cnt));
        memset(bit,0,sizeof(bit));
        for(int i=n-1;i>=0;i--){
            Ri[i]=sum(a[i]-1);
            Ra[i]=n-1-i-Ri[i]-cnt[a[i]];
            add(a[i],1);
            cnt[a[i]]++;
            lb+=Ri[i];
        }
        ll ans=la*lb;
        for(int i=0;i<n;i++){
            ans=ans-Ri[i]*Ra[i];
            ans=ans-Li[i]*La[i];
            ans=ans-Li[i]*Ri[i];
            ans=ans-La[i]*Ra[i];
        }
        printf("%I64d\n",ans);
    }
	return 0;
}