小朋友排队(树状数组)

问题描述

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

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

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

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

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

输入格式

  输入的第一行包含一个整数n,表示小朋友的个数。
  第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。

输出格式

  输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。

样例输入

3
3 2 1

样例输出

9

样例说明

  首先交换身高为3和2的小朋友,再交换身高为3和1的小朋友,再交换身高为2和1的小朋友,每个小朋友的不高兴程度都是3,总和为9。

数据规模和约定

  对于10%的数据, 1<=n<=10;
  对于30%的数据, 1<=n<=1000;
  对于50%的数据, 1<=n<=10000;
  对于100%的数据,1<=n<=100000,0<=Hi<=1000000。

直接做法超时!!

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll s[100000];   //交换次数
ll height[100000];//身高数组
ll n,sum=0;
int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>height[i];
    }
    for(int i=0;i<n;i++)
     for(int j=i;j<n;j++)
     {
      if(height[i]>height[j])
        {
            s[i]++;
            s[j]++;
        }
     }//寻找逆序对
     //标记好交换次数
      for(int i=0;i<n;i++)
      {
        sum=sum+(s[i]+1)*s[i]/2;//累加
      }
      cout<<sum<<endl;
    return 0;
}

说白了就是针对每一个元素,看看它之前有几个大于它的元素,之后有几个小于它的元素,然后记录这个元素在逆序对中出现的次数,也就是含这个元素的逆序对的个数,然后再将每个元素的不高兴程度相加即可。

树状数组实际上是由两部分组成:数据数组+统计数组,只看数据数组(这个描述。。。)

由于树状数组是从1开始的,而题目中小盆友的身高可以为0(真是长见识了),所以我们将每个小盆友的身高加1然后作为树状数组的下标,将数值1存到相应的位置.

第一次读入3,此时读入的数据量为1个,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             0            0            1            0             0             0              0
可以看到sum(C[1],C[4])=1(可以由树状数组的统计数组得到),这个是小于等于3的数字的个数,也就是说当输入第一个数字3的时候没有比它小的数字存在,这时我们用 输入数字总数-sum(C[1],C[4])=0,也就是说大于3的数字的个数为0,我们令b[0]=0.

第二次读入2,此时读入的数据量为2个,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             0            1            1             0                0             0              0

可以看到sum(C[1],C[3])=1,任然不存在比它小的数,但此时输入的数据总量为2,而2-1=1,就是说,存在一个数在2之前并且大于2,这个数当然就是3,我们另b[1]=1.

第三次读入1,此时读入的数据量为3,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             1             1             1              0            0           0            0
可以看到sum(C[1],C[2])=1,任然不存在比它小的数,但此时输入的数据总量为3,而3-1=2,就是说,存在两个数在1之前并且大于1,这个数当然就是2,3,我们另b[2]=2.

到此,我们已经算出了每个数前面的较大的数的个数了,数据存在num[]中,现在我们再反过来遍历,1,2,3,但这次我们不再用总数减去sum了,而直接求sum,求出来的自然就是,每个数后面的较小的数的个数,然后将得到的数值累加到相应的b[i]中,最终我们会得到b[0]=2,b[1]=2,b[2]=2,分别对应num[0]=3,num[1]=2,num[2]=1。

求得了每个小盆友被移动的次数,我们需要计算其不高兴程度,这里实际上可以事先打个表,就是将被移动n次后的不高兴值全算出来,然后直接用就可以了,这里,我们将其存到total[]数组中,而且total[2]=3,所以总不高兴值就是9.


需要注意的是如果重复的数字出现怎么办,如果出现,实际上出问题的会是求每个数之前较大数的那部分,因为用到了数的总个数,如果出现一样的数,就会导致相减后的结果偏大,而且正好是大了 重复量-1 ,那么我们就可以算出重复量,然后将这一部分减去就行,关键是怎么算重复量,实际也很简单,通过树状数组,我们可以求得sum(1,a)和sum(1,a+1),其中输入的数字为a,前者算出的小于a的数的个数,后者算出的是小于等于a的数的个数,两个一减就是等于a的个数。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1111111;
const int N=111111;
ll c1[maxn],c2[maxn],b[maxn];
ll num[N];
ll total[N];
void init()
{
	memset(c1,0,sizeof(c1));
	memset(c2,0,sizeof(c2));
	total[0]=0;
	for(int i=1;i<=N;i++)
        total[i]=total[i-1]+i;
}
int lowbit(int x)
{
    return x&-x;
}
ll query(ll c[],int x)
{
	ll ans=0;
	while(x)
	{
		ans+=c[x];
		x-=lowbit(x);
	}
	return ans;
}
void add(ll c[],int x,ll v)
{
	while(x<maxn)
	{
		c[x]+=v;
		x+=lowbit(x);
	}
}
int main()
{
    ios::sync_with_stdio(false);
    init();
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>num[i];
        add(c1,num[i]+1,1);
        b[i]=i-query(c1,num[i]);
        b[i]=b[i]-(query(c1,num[i]+1)-query(c1,num[i])-1);
    }
    for(int i=n-1;i>=0;i--)
    {
        add(c2,num[i]+1,1);
        b[i]=b[i]+query(c2,num[i]);
    }
    ll ans=0;
    for(int i=0;i<n;i++)
    {
        ans+=total[b[i]];
    }
    cout<<ans<<endl;
    return 0;
}


 

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
树状数组(Fenwick Tree)是一种用于快速维护数组前缀和的数据结构。它可以在 $O(\log n)$ 的时间内完成单点修改和前缀查询操作,比线段树更加简洁高效。 下面是 Java 实现的树状数组详解: 首先,在 Java 中我们需要使用数组来表示树状数组,如下: ``` int[] tree; ``` 接着,我们需要实现两个基本操作:单点修改和前缀查询。 单点修改的实现如下: ``` void update(int index, int value) { while (index < tree.length) { tree[index] += value; index += index & -index; } } ``` 该函数的参数 `index` 表示要修改的位置,`value` 表示修改的值。在函数内部,我们使用了一个 `while` 循环不断向上更新树状数组中相应的节点,直到到达根节点为止。具体来说,我们首先将 `tree[index]` 加上 `value`,然后将 `index` 加上其最后一位为 1 的二进制数,这样就可以更新其父节点了。例如,当 `index` 为 6 时,其二进制表示为 110,最后一位为 2^1,加上后变为 111,即 7,这样就可以更新节点 7 了。 前缀查询的实现如下: ``` int query(int index) { int sum = 0; while (index > 0) { sum += tree[index]; index -= index & -index; } return sum; } ``` 该函数的参数 `index` 表示要查询的前缀的结束位置,即查询 $[1, index]$ 的和。在函数内部,我们同样使用了一个 `while` 循环不断向前查询树状数组中相应的节点,直到到达 0 为止。具体来说,我们首先将 `sum` 加上 `tree[index]`,然后将 `index` 减去其最后一位为 1 的二进制数,这样就可以查询其前一个节点了。例如,当 `index` 为 6 时,其二进制表示为 110,最后一位为 2^1,减去后变为 100,即 4,这样就可以查询节点 4 的值了。 最后,我们还需要初始化树状数组,将其全部置为 0。初始化的实现如下: ``` void init(int[] nums) { tree = new int[nums.length + 1]; for (int i = 1; i <= nums.length; i++) { update(i, nums[i - 1]); } } ``` 该函数的参数 `nums` 表示初始数组的值。在函数内部,我们首先创建一个长度为 `nums.length + 1` 的数组 `tree`,然后逐个将 `nums` 中的元素插入到树状数组中。具体来说,我们调用 `update(i, nums[i - 1])` 来将 `nums[i - 1]` 插入到树状数组的第 `i` 个位置。 到此为止,我们就完成了树状数组的实现。可以看到,树状数组的代码比线段树要简洁很多,而且效率也更高。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值