树状数组BITLA 4329

nuaalida

于 2013-10-02 11:43:16 发布

阅读量535

点赞数

分类专栏：树状数组

本文链接：https://blog.youkuaiyun.com/u011526463/article/details/12233669

版权

树状数组专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用树状数组解决特定问题的方法。通过树状数组实现快速查询和更新操作，有效地解决了与区间和及数据结构相关的问题。文章详细展示了如何初始化树状数组、进行元素添加以及区间求和等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我去，数组最大值跟数据最大值没搞清楚，WA了好多次都看不出来错在哪。这种错误一旦犯了，小数据根本看不不出来。。

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
#define N 100005

int lowbit[N], low[N], a[N];
long long c[N], n;

void Init()
{
    for (int i=1;i<N;i++)
        lowbit[i]=i&(-i);
}

void init()
{
    for (int i=0;i<N;i++)
        c[i]=0;
}

void add(int x,int d)
{
    while (x<N) {
        c[x]+=d;
        x+=lowbit[x];
    }
}

long long sum(int x)
{
    long long js=0;
    while (x) {
        js+=c[x];
        x-=lowbit[x];
    }
    return js;
}

int main()
{
    //freopen("1.in","r",stdin);
    int T, n1;
    long long ans;

    cin>>T;
    Init();
    while (T--){
        scanf("%d",&n1);
        n=n1;
        init();
        for (int i=1;i<=n;i++) {
            scanf("%d",&a[i]);
            low[i]=sum(a[i]);
            add(a[i],1);
        }
        init();
        ans=0;
        add(a[n],1);
        for (int j=n-1;j>1;j--){
            long long i=j, tmp=sum(a[i]);
            ans+=low[i]*(n-i-tmp)+(i-1-low[i])*tmp;
            add(a[i],1);
        }
        cout<<ans<<endl;
    }

    return 0;
}