小朋友排队 解题报告

问题描述

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。

思路：统计每个小朋友前面身高比其高的，以及其后面身高比其矮的，即是该小朋友所要被交换的次数，记为swap[i]

则该小朋友不高兴程度unhappy[i]=swap[i]*(swap[i]+1)/2,然后所有不高兴程度和即unhappy[1]+~~~+unhappy[n],

问题这样转换以后，即可直接套用树状数组模板解决(注意0<=Hi<=1000000这个条件)

先倒着算，求比当前小朋友身高矮的小朋友数目，然后正着算,求比当前小朋友身高高的小朋友的数目

代码如下：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	static int n, step, max;
	static int h[], val[];
	static long ans;
	static long arr[];

	private static int lowbit(int x) {
		return x & (-x);
	}

	private static int sum(int n) {
		int count = 0;
		while (n > 0) {
			count += val[n];
			n -= lowbit(n);
		}
		return count;
	}

	private static void add(int x, int num) {
		while (x <= max) {
			val[x] += num;
			x += lowbit(x);
		}
	}

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner reader = new Scanner(System.in);
		n = reader.nextInt();
		h = new int[n + 1];
		arr = new long[n + 1];
		for (int i = 1; i <= n; i++) {
			h[i] = reader.nextInt();
			h[i]++;
			if (h[i] > max)
				max = h[i];
		}
		val = new int[max + 1];
		for (int i = n; i >= 1; i--) {
			arr[i] += (long) sum(h[i] - 1);
			add(h[i], 1);
		}
		val = new int[max + 1];
		for (int i = 1; i <= n; i++) {
			arr[i] += (long) (sum(max) - sum(h[i]));
			add(h[i], 1);
		}
		for (int i = 1; i <= n; i++)
			ans += arr[i] * (arr[i] + 1) / 2;
		System.out.println(ans);
	}

}

参考（其实还是看了别人题解的，自己一开始只是想到类似冒泡排序的方法~~~）：

http://blog.csdn.net/mymilkbottles/article/details/50688553

关于树状数组(也可参考《算法竞赛入门经典训练指南》)：

http://blog.csdn.net/int64ago/article/details/7429868