小朋友排队问题(树状数组)

小朋友排队问题(树状数组

问题描述

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。

	时间限制:1.0s 内存限制:256.0MB 

分析

统计每位小朋友右边有多少比他高的,最后汇总怒气值
通过树状数组的特性,将身高数组值转为树状数组下标,将每一位都更改为1
对数组进行顺序扫描和反向扫描;

	 通过顺序扫描,更改一位,求出对应的sum值(该sum值表示从第一位数据到该位数据和,在此处用于表示该数据前面有小于等于该数据的元素)
	 通过总添加人数-去该sum得出大于该数据的人数(交换次数)并保存cnt数组
	 
	 因为前些操作导致c数组中数组发生了改变,所以应对c数组再次进行更改 
	 进行反向扫描 
	 最后汇总每位小朋友的交换次数 

c语言代码实现

#include<stdio.h>
#include<string.h> 
int lowbit(int n);
void update(int n,int i,int v,int c[]);
int getsum(int c[],int k); 


int h[100000];			//身高数组 
int c[1000001];			//树状数组,因为要将身高转为下标,身高可能为零,树状数组下标从1开始 
int cnt[100000];		//用于记录每位小朋友的交换次数 	

int main()
{
	int n;
	scanf("%d",&n); 
	
	int maxHit=-1;		//用于保存最大身高 
	for(int i=0;i<n;i++)
	{
		scanf("%d",&h[i]);
		if(h[i]>maxHit) maxHit=h[i]; 
	} 
//	memset(h,sizeof(h));	//初始化 
	memset(c,0,sizeof(c));
	
	//顺序扫描
	for(int i=0;i<n;i++)
	{
		update(maxHit+1,h[i]+1,1,c);		//将树状数组中身高相对应的值更改为1
		int sum = getsum(c,h[i]+1);	//得到该位置的sum值,表示当前小于等于该数据的个数
		cnt[i]+=(i+1)-sum;			//得到,大于该数据的个数,((i+1)表示进入了算当前已经有多少个数据了,而sum表示当前小于该数据的个数) 
	} 
	
	memset(c,0,sizeof(c));			//重新初始化该数据 
	
	//反向扫描
	for(int i=n-1;i>=0;i--)
	{
		update(maxHit+1,h[i]+1,1,c);
		
	//	int sum=getsum(c,h[i]+1);	//小于等于该数据的个数
	//	int self=getsum(c,h[i]+1)-getsum(c,h[i]);	//该数据的个数
	//	int cnt[i]+=sum-self;			//当前数据右侧小于该数据的个数 
		cnt[i]+=getsum(c,h[i]);			//右侧小于该数据的个数 
	} 
	
	//汇总怒气值
	long long ans=0;				//怒气值 
	for(int i=0;i<n;i++)
	{
		ans+=(1+cnt[i])*cnt[i]/2;		//1+2+3+....+n   (1+n)*n/2
	} 
	
	printf("%lli\n",ans);
} 
  
int lowbit(int n) 
{
	return n-(n&(n-1));			//n-1:二进制中变化:消除了源数中二进制中最后一个1
								//(n-1)&n:	彻底去除最后一个1
								//n-(n&(n-1)):保留最后一个1 
} 
  
void update(int n,int i,int v,int c[])		//在k位置增加v更新c数组
{
	//从i位置开始更新,增加lowbit找到下一个位置 
	for(int k=i;k<=n;k+=lowbit(k))
	{
		c[k]+=v;		//在每一与当前位置相关的位置都需要更新 
	} 
	
} 

//求k二进制最后一个1代表的十进制数 

//求前k项和 
int  getsum(int c[],int k)
{
	int sum=0;
	for(int i=k;i>=1;i-=lowbit(i))
	{
		sum+=c[i];
	} 
	
	return sum;
}
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值