题目 1439: 蓝桥杯历届试题-小朋友排队

题目 1439: 蓝桥杯历届试题-小朋友排队

旧题新做，总有不一样的感觉。
不只是会忘记，温故仍可知新。
很简单的题。
用两次树状数组即可。

需要求前面有多少个比它大的，后面有多少个比它小的。

前面比他大的就是逆序对。以往都是数据离散化求，这个题可以不用离散化，因为身高最大才1e6.求逆序对，前 i 个数（包括第i个数）可以和第i个数组成 i 个顺序对或者是逆序对，或者相等。所以，query(a[i]-1)是之前求逆序对，是因为query(a[i]）也包含了自己即求了相等对。所以query(a[i]）包括了自己，相等对。所以用所有的 i - 顺序对和相等对即逆序对。

下面把tree memset一下，逆序add a[i] ，看后面有多少个比他小的。就是直接求顺序对了，直接query(a[i] - 1)即可。

最后的结果保存到一个res数组，最后的结果 = (res[i] + 1) * res[i] / 2*1ll 因为相乘可能超过了int。
还有一点就是输入的身高是从0开始的，树状数组的下标是从0开始的，所有一开始输入的时候要把每个a[i] + 1

世界缤纷美丽。
稻香真的好听。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000000+10;
int tree[N];
int a[N];
int res[N];
int n,m=0;
int lowbit(int x)
{
    return x&-x;
}
void  add(int x,int c)
{
    for(int i=x;i<N;i+=lowbit(i))
        tree[i]+=c;
}
int query(int x)
{
    int ans = 0;
    for(int i=x;i>0;i-=lowbit(i))
        ans+=tree[i];
    return ans;
}
int main()
{
    cin>>n;

    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        ++a[i];
       // m=max(m,a[i]);
        add(a[i],1);
        res[i]+=i-query(a[i]);
    }
    //for(int i=1;i<=n;i++) cout<<res[i]<<" ";cout<<"\n";
    memset(tree,0,sizeof(tree));
    for(int i=n;i>=1;i--)
    {
        add(a[i],1);
        res[i] +=query(a[i] - 1);
    }
    long long ans = 0;
    for(int i=1;i<=n;i++)
    {
      //  cout<<res[i]<<" ";
        ans += 1ll*(res[i]+1)*res[i]/2;
    }
    cout<<ans;
}