Codeforces Round #609 Div2

1269E K Integers（数据结构）

题意：

​ 给一个1-n的排列，每进行一次操作可交换序列中相邻的两数，设$f(k)$为使序列出现连续的1，2，…，k所需要的最小操作数，求所有的$f(k),k=1,2,...,n(n\le 2\ast 10^5)$。

解法：

​ 当一段长度为m的子序列为一个1-m的排列时，显然$f(m)$为子序列中逆序对总个数。于是进一步考虑求$f(k)$时，将1-k这k个数字”聚拢“成连续的一段所需要的操作数。可以证明，以位于最中间的数字为“中心”聚拢时交换次数最小。设1-k在原序列中的位置为$p_1,p_2,...,p_k$，中心位置为$l$，$r=l+1$，整个序列被分为两部分：$[1,l]、[r,n]$。设位于$[1,l]$的$p_i$有$lcnt$个，位于$[r,n]$的$p_i$有$rcnt$个，则将位于$[1,l]$的$p_i$交换至$[l-lcnt+1,l]$，交换次数为$l\ast lcnt-\sum _{p_i\le l}{p}_i-(1+2+...+lcnt-1)$，同理对$[r,n]$中的$p_i$交换次数为$\sum _{p_i\ge r}{p}_i-r\ast rcnt-(1+2+...+rcnt-1)$。

​ 综上，我们只需在k从1到n递增的过程中动态维护$l、lcnt、\sum _{p_i\le l}{p}_i$。利用一个大根堆和一个小根堆可实现动态维护序列$p_i$的中位数$l$。再分别用两个树状数组$tree1、tree2$维护$lcnt$和$\sum _{p_i\le l}{p}_i$，修改时在位置$p_i$上，$tree1$每次增加1，$tree2$每次增加$p_i$。求逆序对数利用$tree1$即可。

​ 复杂度：$O(nlogn)$。

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) ((x)&(-x))
using namespace std;
const int maxn=2e5+5;
int n,p[maxn],idx[maxn];
ll tree1[maxn],tree2[maxn];
priority_queue<int>lhp;
priority_queue<int,vector<int>,greater<int> >rhp;
void update(int x,int val)
{	while(x<=n) tree1[x]++,tree2[x]+=val,x+=lowbit(x); }
ll query(int x,ll* tree)
{
	ll ans=0;
	while(x) ans+=tree[x],x-=lowbit(x);
	return ans;
}
void insert(int x)
{
	if(!rhp.empty()&&x>=rhp.top())
	{
		lhp.push(rhp.top());
		rhp.pop(),rhp.push(x);
	}
	else lhp.push(x);
	if(rhp.size()+1<lhp.size())
		rhp.push(lhp.top()),lhp.pop();
}
ll sum(ll x)
{	return x*(x+1)/2; }
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>p[i],idx[p[i]]=i;
	ll inv=0;
	for(int i=1;i<=n;i++)
	{
		update(idx[i],idx[i]),insert(idx[i]);
		int l=lhp.top(),r=l+1;
		int lcnt=query(l,tree1),rcnt=i-lcnt;
		inv+=i-query(idx[i],tree1);
		ll ans=inv+query(n,tree2)-2*query(l,tree2);
		ans+=l*lcnt-r*rcnt-sum(lcnt-1)-sum(rcnt-1);
		cout<<ans<<" ";
	}
}