蓝桥 - 小朋友排队 树状数组

思路:树状数组处理逆序对
求前面有多少个数比当前这个大:正着插
求后面有多少个数比当前这个小:反着插
求出f数组和b数组即可
下面是AC代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define sd(a) scanf("%d",&a)
#define sdd(a,b) scanf("%d%d",&a,&b)
#define cl(a,b) memset(a,b,sizeof(a))
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define sddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define dbg() printf("aaa\n")
using namespace std;
//要处理出 前面比这个数大的 和后面比这个数小的数
const int maxn=1e5+10;
int n;
int a[maxn],c[maxn],acopy[maxn];
int f[maxn],b[maxn];//f 前面几个数比它大  b 后面几个数比它小
inline int lb(int x){
    return x&-x;
}
void add(int i,int x){
    while(i<=n){
        c[i]+=x;
        i+=lb(i);
    }
}
int getsum(int i){//获得从1到i的数量
    int ans=0;
    while(i>=1){
        ans+=c[i];
        i-=lb(i);
    }
    return ans;
}
int main() {
	sd(n);
    rep(i,1,n) sd(a[i]);
    rep(i,1,n) acopy[i]=a[i];
    sort(acopy+1,acopy+n+1);
    int size=unique(acopy+1,acopy+n+1)-(acopy+1);
    //刚开始思路错了!!!应该是按坐标放!

    //到这已经离散化好了
    //要先找每个坐标前面有多少比它大的
    //f[i]=getsum()
    //c[i]统计的是  不大于i的数有几个
    for(int i=1;i<=n;i++){//先统计
        a[i]=lower_bound(acopy+1,acopy+size+1,a[i])-acopy;
        f[i]=getsum(n)-getsum(a[i]);
        add(a[i],1);
    }
    cl(c,0);
    for(int i=n;i>=1;i--){
        b[i]=getsum(a[i]-1);
        add(a[i],1);
    }
    ll ans=0;
    rep(i,1,n){//这里会爆ll!!!
        ans+=(1ll+(ll)f[i]+(ll)b[i])*((ll)f[i]+(ll)b[i])/2;
    }
    printf("%lld\n",ans);
	return 0;
}

题目描述
n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。

每个小朋友都有一个不高兴的程度。开始的时候,所有小朋友的不高兴程度都是0。

如果某个小朋友第一次被要求交换,则他的不高兴程度增加1,如果第二次要求他交换,则他的不高兴程度增加2(即不高兴程度为3),依次类推。当要求某个小朋友第k次交换时,他的不高兴程度增加k。

请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。

如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。

样例说明
首先交换身高为3和2的小朋友,再交换身高为3和1的小朋友,再交换身高为2和1的小朋友,每个小朋友的不高兴程度都是3,总和为9。
数据规模和约定
对于100%的数据,1< =n< =100000,0< =Hi< =1000000。

输入
输入的第一行包含一个整数n,表示小朋友的个数。
第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。
输出
输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。
样例输入

3
3 2 1

样例输出

9

发布了120 篇原创文章 · 获赞 12 · 访问量 5262

猜你喜欢

转载自blog.csdn.net/weixin_43735161/article/details/104992388