【小朋友排队】

Kent_J_Truman

已于 2024-09-02 16:37:49 修改

阅读量100

点赞数

分类专栏：蓝桥杯爆int 算法文章标签：算法蓝桥杯树状数组

于 2024-09-02 16:36:28 首次发布

本文链接：https://blog.csdn.net/m0_73669127/article/details/141822770

版权

算法同时被 3 个专栏收录

75 篇文章 0 订阅

订阅专栏

蓝桥杯

36 篇文章 2 订阅

订阅专栏

爆int

2 篇文章 0 订阅

订阅专栏

`题目`

在这里插入图片描述

`代码`

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10, M = 1e6+10;
int a[N], cnt[N];
long long t[M]; 
int n;
int lowbit(int x)
{
    return x & -x;
}
void update(int x, int d)
{
    for(; x < M; x += lowbit(x)) t[x] += d;
}
int query(int x)
{
    int retval = 0;
    for(; x >= 1; x -= lowbit(x)) retval += t[x];
    return retval;
}
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++) cin >> a[i], a[i]++;
    for(int i = 1; i <= n; i++)
    {
        update(a[i], 1);
        cnt[i] += i - query(a[i]);
    }
    memset(t, 0, sizeof t);
    for(int i = n; i >= 1; i--)
    {
        update(a[i], 1);
        cnt[i] += query(a[i] - 1);
    }
    
    long long res = 0;
    for(int i = 1; i <= n; i++) res += (1 + cnt[i]) * (long long)cnt[i] / 2;
    
    cout << res;
    
    return 0;
}