【Luogu】 P4331 [BalticOI 2004] Sequence 数字序列

题目链接

点击打开链接

题目解法

首先做一个重要的转化：把 $b_i$ 单调上升变为 $b_i$ 单调不降
如何转化？将 $a_i-i$ 变成新的 $a_i$，将 $b_i-i$ 变新的 $b_i$，这样答案是不变的，且 $b_i$ 变成了单调不降

考虑对于两段相邻序列 $a_1,...,a_n$ 和 $a_{n+1},...,a_{n+m}$，最优的一组 $b$ 分别为 $b_1,...,b_n(b_1=b_2=...=b_n)$ 和 $b_{n+1},...,b_{n+m}(b_{b+1}=b_{n+2}=...=b_{n+m})$
考虑合并之后的 $b_1,...,b_{n+m}$
令 $b_1=u,b_{n+1}=v$

1. $u\le v$，$b_i$ 的值不用改变，直接合并即可
2. $u>v$
   令合并后的答案为 $c_1,...,c_{n+m}$
   考虑结论：$c_n\le u$
   证明：如果有 $c_n>u$，那么把 $c_1,...,c_n$ 替换成 $u$ 也是一组合法的解，且答案不会变劣
   同理可得 $c_{n+1}\ge v$
   考虑最优解 $c_1,...,c_{n+m}(c_n\le u,c_{n+1}\ge v)$
   把 $c_1,...,c_{n+m}$ 替换为 $a_1,...,a_{n+m}$ 的中位数 $k$ 答案不会更劣
   证明：若 $c_1>k$ 或 $c_{n+m}<k$，那么易知可以替换；若 $c_n<k$ 且 $c_{n+1}>k$，根据 $1-n$ 中 $\ge u$ 的 $a_i$ 一定不小于一半，$n+1-m$中 $lev$ 的 $a_i$ 一定不小于一半可以证得

所以可以一个一个添加数，一段一段往前合并即可

考虑如何合并？用左偏树维护前一半的最大值
考虑合并如何合并两个左偏树？直接合并即可
直接合并左偏树维护中位数在没有限制的两个序列中是不对的
但这道题中是有特殊限制的
考虑新的中位数一定出现在橘色框内，且上面序列的橘色框是维护到了，但下面序列的橘色框内是没有维护到的，但因为只添加了一个数，且之前的中位数是大于上方序列的中位数的，所以下面序列的橘色框内没有数，所以直接合并恰好是可行的

时间复杂度 $O(nlogn)$

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N(1000100);
struct Node{
	int ed,rt,siz;
}stk[N];
int n,a[N],ans[N];
int lc[N],rc[N],dist[N],v[N];
inline int read(){
	int FF=0,RR=1;
	char ch=getchar();
	for(;!isdigit(ch);ch=getchar()) if(ch=='-') RR=-1;
	for(;isdigit(ch);ch=getchar()) FF=(FF<<1)+(FF<<3)+ch-48;
	return FF*RR;
}
int merge(int x,int y){
	if(!x||!y) return x|y;
	if(v[x]<v[y]) swap(x,y);
	rc[x]=merge(rc[x],y);
	if(dist[rc[x]]>dist[lc[x]]) swap(lc[x],rc[x]);
	dist[x]=dist[rc[x]]+1; 
	return x;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++) a[i]=read(),a[i]-=i;
	int top=0;
	for(int i=1;i<=n;i++){
		v[i]=a[i],dist[i]=1;
		Node t={i,i,1};
		while(top&&v[t.rt]<v[stk[top].rt]){
			t.rt=merge(t.rt,stk[top].rt);
			if((t.siz&1)&&(stk[top].siz&1)) t.rt=merge(lc[t.rt],rc[t.rt]);
			t.siz+=stk[top].siz;
			top--;
		} 
		stk[++top]=t;
	}
	for(int i=1,j=1;i<=n;i++){
		if(i>stk[j].ed) j++;
		ans[i]=v[stk[j].rt];
	}
	LL tot=0;
	for(int i=1;i<=n;i++) tot+=abs(ans[i]-a[i]);
	printf("%lld\n",tot);
	for(int i=1;i<=n;i++) printf("%d ",ans[i]+i);
	return 0;
}