Luogu P6186 [NOI Online 提高组]冒泡排序

题面

大意就是给定一个序列，对其进行两个操作，交换相邻的两个数，或者对全序列进行一遍冒泡排序。

观察题面可以发现

• 当ti=1时我们需要交换相邻的两个数
• 当ti=2时我们需要对全序列进行冒泡排序

由于数据量极大，显然暴力的模拟一定不行

我们记录第i位数前面比它大的数的数量为$before[i]$，显然，当前序列的总逆序对数量就是所有的$before$之和

通过对冒泡排序的观察，我们可以发现，每一遍冒泡排序都会使得所有$before[i]=max(before[i]-1,0)$

我们采用树状数组差分维护这一操作，令$sum(t)$为当$ti=2$，$k=t$时的答案

在数组最前面加入当前序列总逆序对数量，然后在第$i$位放$before$大于$i$的数字的数量的相反数，因为这些数字在第$i$轮逆序对数均会减$1$

最后利用差分直接求解即可

#include <bits/stdc++.h>
using namespace std;

int input[200005], before[200005], record[200005];
int n;

long long tree[200005];
inline int lowbit(int x)
{
	return x & (-x);
}
void add(int x, long long val)
{
	for (; x <= n; x += lowbit(x))
		tree[x] += val;
	return;
}
long long sum(int x)
{
	long long res = 0;
	for (; x > 0; x -= lowbit(x))
		res += tree[x];
	return res;
}

int main()
{
	int m;
	scanf("%d%d", &n, &m);
	long long tot = 0; //先记录初始状态下的逆序对数量
	for (int i = 0; i < n; i++)
	{
		scanf("%d", input + i);
		before[i] = i - sum(input[i]); //记录比它小的数字数量
		tot += before[i];			   //最开始tot记录初始的答案
		record[before[i]]++;		   //桶，record[i]=前面有i个数字比它大的数字的数量
		add(input[i], 1);			   //树状数组作桶
	}
	memset(tree, 0, sizeof(tree)); //清空
	add(1, tot);				   //实现差分，先把序列总逆序对数量放在最前面
	tot = 0;
	for (int i = 0; i < n; ++i)
	{
		tot += record[i]; //每次tot记录的是前面有小于等于i个数字比它大的数字的数量
		//则n-tot即为前面有大于i个数字比它大的数字的数量
		add(i + 2, -(n - tot)); //实现差分，在这一个位置会有n-tot个数字逆序对数减1
								//由于下标问题，i必须+2，这样当i=0时就会储存在第2位，而第1位是放总逆序对数的
	}
	for (int i = 0, opt, x; i < m; i++)
	{
		scanf("%d%d", &opt, &x);
		x = min(x, n - 1); //对opt=2的情况进行优化
		if (opt == 1)
		{
			x--;
			if (input[x] < input[x + 1])
			{
				swap(input[x], input[x + 1]);
				swap(before[x], before[x + 1]);
				add(1, 1);					//逆序对总数量增加1
				add(before[x + 1] + 2, -1); //由于before[x+1]增加了，所以在原before[x+1]轮时也可以使逆序对-1，所以记录-1
				before[x + 1]++;			//input[x]交换到x+1位上后，前面比它大的数量增加了1
			}
			else
			{
				swap(input[x], input[x + 1]);
				swap(before[x], before[x + 1]);
				add(1, -1);			   //逆序对总数量减少1
				before[x]--;		   //input[x+1]交换到x位上后，前面的比它大的数量减少了1
				add(before[x] + 2, 1); //由于before[x]减少了，所以在原before[x]轮时无法使逆序对减少，所以记录1
			}
		}
		else
			printf("%lld\n", sum(x + 1)); //直接输出答案
	}
	return 0;
}