蓝桥杯 小朋友排队（线段树，树状数组求逆序数）

历届试题 小朋友排队  

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

    

问题描述

　　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。

tips:复习了下线段树和树状数组，用了两种方法都实现了下。

感觉注意几点：

1.清楚身高的上界值和人数的上界值。例如sum数组和c数组都是看的身高上界的最大值。

2线段树中sum数组代表【L,R】区间内的数字出现的次数。

3.树状数组中c【i】数组代表的是区间【1，i】内的数字出现的次数、

4.线段树写起来，难度要大，但是思路上个人觉得要更清晰，具体：1）线段树可以直接找前面比当前身高高的，不需要树状数组一次减去相同身高的。

另外1）注意身高不能最小为0，因为在减1的时候可能出现溢出。2）为了防止出现区间的右端点小于左端点，我们可以在query（）函数里进行特判，或者建树的时候左端点设置为0.

5.另外存储结果需要使用long long

具体实现要写的太多，附上几位大神的讲解博客：http://blog.csdn.net/wr132/article/details/43856905

http://blog.csdn.net/strokess/article/details/51204402感谢鲍总！不过觉得鲍总解释里的sum数组怪怪的。

代码：

1）线段树

#include<iostream>
#include<cstring>
#include<fstream>
using namespace std;
const int maxn=100000+1000;
const int maxh=1000000+100;
int sum[maxh<<2],a[maxn];
int rev[maxn],n;
long long ans;
/*
void build(int rt,int l,int r)
{
	if(l==r)
	{
		sum[rt]=0;
		return;
	}
	int m=(l+r)>>1;
	build(rt<<1,l,m);
	build(rt<<1|1,m+1,r);
	pushup(rt);
}
*/
void pushup(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int x,int add,int rt,int l,int r)
{
	if(l==r)
	{
		sum[rt]++;
		return;
	}
	int m=(l+r)>>1;
	if(x<=m)update(x,add,rt<<1,l,m);
	else update(x,add,rt<<1|1,m+1,r); 
	pushup(rt); 
}
int query(int ll,int rr,int rt,int l,int r)
{
	//if(rr<ll)return 0; //特判 （有可能出现右端点小于左端点的情况） 
	if(ll<=l&&rr>=r)
	{
		return sum[rt];
	}
	int ret=0;
	int m=(l+r)>>1;
	
	if(ll<=m)ret+=query(ll,rr,rt<<1,l,m);
	if(rr>m)ret+=query(ll,rr,rt<<1|1,m+1,r); 
	return ret;
}
int main()
{
	scanf("%d",&n);
	//build(1,0,maxh);
	//ifstream input("input.txt");
	//input>>n;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	//	input>>a[i];
		update(a[i]+1,1,1,0,maxh);
		rev[i]=query(a[i]+2,maxh,1,0,maxh);//找前面比当前身高高的
	//	rev[i]=i-query(0,a[i],1,0,maxh)-1;
	//	rev[i]-=query(0,a[i]+1,1,0,maxh)-query(0,a[i],1,0,maxh)-1;
	}
	memset(sum,0,sizeof(sum));
	for(int i=n;i>=1;i--)
	{
		update(a[i]+1,1,1,0,maxh);
		rev[i]+=query(0,a[i],1,0,maxh); //找后面比当前身高矮的
	}
	
	for(int i=1;i<=n;i++){
		ans+=((long long) rev[i]*(rev[i]+1))/2;
	}
	cout<<ans<<endl;
	return 0;
}




2)树状数组


#include<iostream>
#include<cstring>
#include<fstream>
using namespace std;
#define ll long long 
const int maxn=100000+1000;
const int maxh=1000000+1000;
int c[maxh],a[maxn];
int rev[maxn],n;
ll ans;
int lowbit(int x)
{
	return x&(-x);
}
void update(int i,int add)
{
	while(i<=maxh)
	{
		c[i]+=add;
		i+=lowbit(i);
	}
}
int query(int i)
{
	int cnt=0;
	while(i>=1)
	{
		cnt+=c[i];
		i-=lowbit(i);
	}
	return cnt;
 } 
 ll cal(int x)
 {
 	return (ll)x*(x+1)/2; 
 }
int main()
{
	 
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		
		scanf("%d",a+i);
		update(a[i]+1,1);
		rev[i]=i-(1+query(a[i]));//用当前所有人-（他自己+比他小的所有人）==大于等于他的所有人（不包括他自己） 
		rev[i]-=query(a[i]+1)-query(a[i])-1;//右边的部分是 （小于等于自己的-减去小于自己的-自己==和自己身高相同的） 

	}
	//计算所有后面比他小的 
	memset(c,0,sizeof(c));
	for(int i=n;i>=1;i--)
	{
		update(a[i]+1,1);
		rev[i]+=query(a[i]);
	}
	
	for(int i=1;i<=n;i++)
	{
		ans+=cal(rev[i]);
	}
	cout<<ans<<endl;
	return 0;
 }