试题 历届试题 小朋友排队

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
pair<int,int>a[maxn];
int dis[maxn];
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        scanf("%d",&a[i].first);
        a[i].second=0;
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n-i-1;j++){
            if(a[j].first>a[j+1].first){
                a[j].second++;
                a[j+1].second++;
                dis[j]+=a[j].second;
                dis[j+1]+=a[j+1].second;
                swap(a[j+1],a[j]);
            }
        }
    }
    long long int ans=0;
    for(int i=0;i<n;i++){
        ans+=dis[i];
    }
    cout<<ans<<endl;
    return 0;
}

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
typedef long long int ll;
int tree[maxn];
ll sum[maxn];
int hi[maxn];
int n,l;
int lowbit(int x){
    return x&(-x);
}
void add(int x,int t){
    for(int i=x;i<maxn;i+=lowbit(i))tree[i]+=t;
}
ll query(int x){
    int res=0;
    for(int i=x;i>0;i-=lowbit(i))res+=tree[i];
    return res;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d",&hi[i]);
        hi[i]++;
    }
    for(int i=1;i<=n;i++){///前面比你大的数，从首项开始计算
        sum[i]+=query(maxn)-query(hi[i]);///得到的是前面比当前数大的数
        add(hi[i],1);///以值作为树状数组的下标
    }
    fill(tree,tree+maxn,0);///清空
    for(int i=n;i>=1;i--){///后面比你小的数，从尾项开始计算
        sum[i]+=query(hi[i]-1);
        add(hi[i],1);
    }
    ll ans=0;
    for(int i=1;i<=n;i++){///na1+n(n-1)/2*d=n*(n+1)/2; a1=1;sum[i]=n;
        ans+=sum[i]*(sum[i]+1)/2;///等差数列，前n项和化简
    }
    cout<<ans<<endl;
    return 0;
}