题目
论文题。
好难啊。
我们先考虑b数组不下降。
那么一定是一整条一整条的递增。。。。。。
现在如果前半段和后半段的最优决策如此:
那么整段的最优决策也就这样了。
但是如果是这样的:
那么不能直接合并。
首先可以证明,整个的最优策略中,左边还是一条直线,右边也是。
然后可以证明,这两条直线应该是一个高度(左边的想往上,右边的想往下,合在一起)。
然后可以发现,这个高度就是中位数的高度。。。。。。
所以就在后面加点,和前面一直判断合并,维护中位数,因为此题合并的条件保证中位数不增,我们可以用大根可并堆来维护中位数。。。。。
如果必须递增,那么我们就把每个数-它的下标当作它新的值。
这样原来的方案可以和不下降的方案一一对应,
问题被转化为b数组不下降。
好难啊。
AC Code:
#include<bits/stdc++.h>
#define maxn 1000005
#define LL long long
using namespace std;
int n;
int rt[maxn],ch[maxn][2],val[maxn],l[maxn],siz[maxn],d[maxn];
int sta[maxn],tp,b[maxn];
void merge(int &now,int l,int r){
if(!l || !r){ now = l + r;return;}
if(val[l] > val[r]) now = l , merge(ch[now][1],ch[l][1],r);
else now = r , merge(ch[now][1],l,ch[r][1]);
if(d[ch[now][1]] > d[ch[now][0]]) swap(ch[now][0],ch[now][1]);
d[now] = d[ch[now][1]] + 1;
siz[now] = siz[ch[now][0]] + siz[ch[now][1]] + 1;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&val[i]);val[i] -= i;
rt[i] = l[i] = i;
siz[i] = 1;
for(;tp && val[rt[sta[tp]]]>val[rt[i]];tp--){
merge(rt[i],rt[sta[tp]],rt[i]);
l[i] = l[sta[tp]];
//printf("%d %d %d\n",i,val[rt[i]],siz[rt[i]]);
while(siz[rt[i]] > (i-l[i]+2)/2) merge(rt[i],ch[rt[i]][0],ch[rt[i]][1]);
//printf("%d %d %d\n",i,val[rt[i]],siz[rt[i]]);
}
sta[++tp] = i;
}
LL ans = 0;
for(;tp;){
for(int i=l[sta[tp]];i<=sta[tp];i++)
b[i] = val[rt[sta[tp]]] , ans+=abs(val[i] - b[i]);
tp--;
}
printf("%lld\n",ans);
for(int i=1;i<=n;i++)
printf("%d%c",b[i]+i,i==n?'\n':' ');
}