蓝桥杯--树状数组2 AcWing 1215. 小朋友排队

AcWing 1215. 小朋友排队

n 个小朋友站成一排。
现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。
每个小朋友都有一个不高兴的程度。
开始的时候，所有小朋友的不高兴程度都是 0。
如果某个小朋友第一次被要求交换，则他的不高兴程度增加 1，如果第二次要求他交换，则他的不高兴程度增加 2（即不高兴程度为 3），依次类推。当要求某个小朋友第 k 次交换时，他的不高兴程度增加 k。
请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。
输入格式
输入的第一行包含一个整数 n，表示小朋友的个数。
第二行包含 n 个整数 H1,H2,…,Hn，分别表示每个小朋友的身高。
输出格式
输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。
数据范围
1≤n≤100000,
0≤Hi≤1000000
输入样例：
3
3 2 1
输出样例：
9

题意：给出身高序列，每次交换相邻两人使序列整体升序，每个小朋友被交换第n次时怒气值加n，求所有人怒气值和的最小值

首先难点在于分析出如何求出交换的最小次数，需要交换也就说明序列中存在逆序对。对于第x个小朋友，设其前方有k1个身高高于h[x]，其后方有k2个身高小于h[x]，则对于x来说，存在k1+k2个逆序对，即使在最优情况下，每次调整也只能消除一个逆序对，因此对于x来说k1+k2个逆序对至少要交换k1+k2次

那么如何确定何时能够取到最少交换次数k1+k2，个人理解是对于某个小朋友x当在k1+k2次交换中，如果有一次交换可以直接消除一个逆序对，并且这次交换不会产生一个新的逆序对，那么这次交换就是最优操作的中的一步，可以将其执行。

当这次交换结束后，如果对于x来说剩余的交换次数中，不存在这样的一步，可以转移目标，观察其他小朋友y，对y来进行类似操作，更新序列，可能会产生允许x进行的操作，因为存在最小值，所以不会出现无解的情况

如序列4 2 5 1 3
先统计出身高为4的小朋友k1值为0，k2值为3，则最少交换次数为3
观察4 2，如果进行这次操作，则会减少一个4 2逆序对，同时对于其他元素而言，也不会产生新的逆序对，那么这次操作是可执行的且最优的
序列更新为2 4 5 1 3，身高为4和2小朋友的交换次数减一，继续执行类似操作

那么对于x来说，如果其前方存在k1个大于他的个数，后方存在k2个小于他的个数，每次都进行最优解的情况下，只需进行k1+k2次操作即可将其转移到目标位置，且k1+k2位最小交换数，那么对于整个序列来说，计算出每个小朋友的k1+k2的值，进行怒气值计算即可

对于k1k2的获得，并不需要在意具体操作顺序如何，只需知道这种方式存在即可，可以模拟一遍操作的流程便于理解

至此明确出问题转化为计算每个小朋友的k1+k2值，这里则运用到了树状数组，避免使用遍历带来的超时，使用树状数组tr[x]来记录身高为x的个数，通过query方法来求取区域和

注意的是题目中小朋友身高会取到0，而树状数组要从1开始，因此将身高+1，并不影响最终结果

代码如下

import java.io.*;
import java.util.*;

public class Main {
	static Scanner tab = new Scanner(System.in);
	static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	static int N = 1000010;

	static int tr[]=new int [N];
	static int h[]=new int [N]; //身高记录
	static int sum[]=new int [N];//交换次数记录
	static int n;
	
	public static int lowbit(int x) {
		return x&-x;
	}
	
	public static int query(int x) {
		int res=0;
		for(int i=x;i>0;i-=lowbit(i)) {
			res+=tr[i];
		}
		return res;
	}
	
	public static void add(int x,int v) {
		for(int i=x;i<N;i+=lowbit(i)) {
			tr[i]+=v;
		}
	}
	
	public static void main(String[] args) throws IOException {		
		n=tab.nextInt();
		for(int i=0;i<n;i++) {
			h[i]=tab.nextInt();
			h[i]++;
		}
		
		//前半大于h[i]的个数
		for(int i=0;i<n;i++) {
			sum[i]=query(N-1)-query(h[i]);
			add(h[i], 1);
		}
		Arrays.fill(tr, 0);//重置树状数组
		
		//后半小于h[i]的个数
		for(int i=n-1;i>=0;i--) {
			sum[i]+=query(h[i]-1);
			add(h[i], 1);
		}
		
		long res=0;//防止爆int
		for(int i=0;i<n;i++) {
			res +=(long)sum[i] * (sum[i] + 1) / 2;
		}
		System.out.println(res);
	}
}

---

比较抽象，还是建议纸上模拟一遍比较好理解