树状数组专题(四)-数列-训练套题T10T3

最新推荐文章于 2021-03-07 15:44:32 发布

WUST_XIAO

最新推荐文章于 2021-03-07 15:44:32 发布

阅读量255

点赞数 1

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/xr469786706/article/details/88068509

版权

数据结构与算法专栏收录该内容

67 篇文章 1 订阅

订阅专栏

树状数组专题(四)-数列-训练套题T10T3

在这里插入图片描述

代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 50000+5;
const int maxm = 32771;
int a[maxn];
int c[maxn];
int tmp[maxn];
int n;
int lowbit(int x){
    return x & (-x);
}

void update(int x,int v){
    for(int i = x ; i < maxn ; i += lowbit(i)){
        c[i] += v;
    }
}

int getSum(int x){
    int sum = 0;
    for(int i = x ; i > 0 ; i -= lowbit(i)){
        sum += c[i];
    }
    return sum;
}

int main(){
    freopen("in.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        long long ans = 0;
        memset(c,0,sizeof(c));
        for(int i = 1 ; i <= n ; i++){
            scanf("%d",&a[i]);
        }
        for(int i = 1 ; i <= n ; i++){
            update(a[i],1);
            tmp[i] = getSum(a[i]-1);
        }
        memset(c,0,sizeof(c));
        for(int i = n ; i >= 1 ; i--){
            update(a[i],1);
            ans += tmp[i]*getSum(a[i]-1);
        }
        printf("%lld",ans);
    }


    return 0;
}