历届试题 小朋友排队
时间限制:1.0s 内存限制:256.0MB
问题描述
n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。
每个小朋友都有一个不高兴的程度。开始的时候,所有小朋友的不高兴程度都是0。
如果某个小朋友第一次被要求交换,则他的不高兴程度增加1,如果第二次要求他交换,则他的不高兴程度增加2(即不高兴程度为3),依次类推。当要求某个小朋友第k次交换时,他的不高兴程度增加k。
请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。
每个小朋友都有一个不高兴的程度。开始的时候,所有小朋友的不高兴程度都是0。
如果某个小朋友第一次被要求交换,则他的不高兴程度增加1,如果第二次要求他交换,则他的不高兴程度增加2(即不高兴程度为3),依次类推。当要求某个小朋友第k次交换时,他的不高兴程度增加k。
请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。
输入格式
输入的第一行包含一个整数n,表示小朋友的个数。
第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。
第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。
输出格式
输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。
样例输入
3
3 2 1
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。
对于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; }