洛谷P6278题解

洛谷P6278题解

题目描述

Farmer John 由于对整理他难以整平的头发感到疲惫，于是决定去理发。他有一排 $N$缕头发，第 ii 缕头发初始时长度为 $A_i$微米$（0\le A_i\le N）。$理想情况下，他想要他的头发在长度上单调递增，所以他定义他的头发的“不良度”为逆序对的数量：满足 $i<j$及 $A_i>A_j$的二元对 $(i,j)$。
对于每一个$j=0,1,\dots ,N-1$，Farmer John 想要知道他所有长度大于j的头发的长度均减少到 j时他的头发的不良度。

这道题真的卡了我好长时间，本来想拿这道题来学习树状数组求逆序对的，结果看了题解被绕了半天，不得不说，洛谷里这道题的题解说的都不清楚，不适合树状数组求逆序对的入门选手，下面写一篇详细一点的。

首先，这道题需要用到一个比较巧妙的思想：正难则反。既然正着减少不好想，那么放过来增加是好想的对吧。于是，这道题就变成了原来所有头发都是$0$，每个时刻所有头发增加$1$，第$i$根头发最大长度为$A_i$，求每个时刻逆序对的个数。如果能求出逆序对，那么答案就是不断累加，因为当前时刻的逆序对总数 = 之前所有时刻的逆序对总数 + 该时刻新增的逆序对个数。

下面看怎么求这个逆序对。回顾最直白的逆序对我们怎么求的：先离散化（取排序后的下标），然后再用树状数组维护离散化后的数组。其实我的理解是，这个离散化无非是想找一种有序的标准，能够满足大的数先被加入树状数组（即越小的越大），这样的话我们访问前缀和就能获得比当前这个数大的数的数量。好了，回到这道题，我们怎么找这个标准呢？答案是用$n-A_i$，因为这个量表示第$i$根头发与最大高度的差距，这个值越小，说明这根头发越大。于是这道题就解决了，我们用$sum[A_i]$表示$A_i$时刻新增的逆序对数量，而求法就是标准的树状数组写法。

代码中之所以都+1是因为求逆序对的时候树状数组不能维护0，因此我们把整体范围都扩大1，不影响答案。

#include<cstdio>
#include<iostream>
#include<cstring>
#define N 500005
#define ll long long
using namespace std;
int n, a[N], c[N];
ll sum[N], ans;

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

void add(int x, int k) {
	for(int i = x; i <= n + 1; i += lowbit(i)) {
		c[i] += k;
	}
}

ll query(int x) {
	ll res = 0;
	for(int i = x; i; i -= lowbit(i)) {
		res += c[i];
	}
	return res;
}

int main()
{
	cin>>n;
	for(int i = 1; i <= n; ++i) {
		cin>>a[i];
		a[i]++;
	}
	for(int i = 1; i <= n; ++i) {
		int x = n - a[i] + 1;
		sum[a[i]] += query(x);
		add(x + 1, 1);
	}
	cout<<"0"<<endl;
	for(int i = 1; i < n; ++i) 
	{
		ans += sum[i];
		cout<<ans<<endl;
	}
	return 0;

看完代码后你也许又有一个疑惑：那$A$数组肯定不会正好包含$1\sim n+1$，那些没出现过的$sum$值怎么办？

答案是没有问题。我们来考虑两个数$A_i,A_j$，假设这本来是一个逆序对，即$i<j,A_i>A_j$，那么对于任意的$k$，如果说$A_j<k$，那么很显然$A_i$变成$k$以后不影响这个逆序对，因此答案不变；如果$A_j\ge k$，那么答案确实会变化，但是按照我们反着做的做法，$A_j>k$的情况其实会被$A_j=k$的情况覆盖，因此我们只需要考虑$A_j=k$的情况即可。