Luogu P4331 [BOI2004]Sequence 数字序列
思维题OrzOrz;
为了方便思考(想不到),我们可以把 \(a_i\) 减去 \(i\) ,把严格递增的条件转化为非严格递增的条件。
首先如果 \(a\) 递增,我们取 \(b_i=a_i\) 即可;若 \(a\) 递减,我们取 \(b\) 为 \(a\) 的中位数即可(均分纸牌)。
而这道题可以运用类似的方法:我们先把每一个元素当做独立的一段,即假设 \(b_i=a_i\) ;若 \(b_{i-1}\leq b_i\) 则是合法的;若 \(b_{i-1}> b_i\) 则我们把 \(i\) 扔进上一段即可;于是会产生很多段(此时不一定),我们只需要用可并堆维护中位数即可。
#include<iostream>
#include<cstdio>
#include<cmath>
#define R register int
#define ll long long
using namespace std;
namespace Luitaryi {
inline int g() { R x=0,f=1;
register char s; while(!isdigit(s=getchar())) f=s=='-'?-1:f;
do x=x*10+(s^48); while(isdigit(s=getchar())); return x*f;
} const int N=1000010;
int n; int ans[N]; ll anss;
int ls[N],rs[N],d[N],vl[N];
inline int merge(int x,int y) {
if(!x||!y) return x+y;
if(vl[x]<vl[y]) swap(x,y);
rs[x]=merge(rs[x],y);
if(d[ls[x]]<d[rs[x]]) swap(ls[x],rs[x]);
d[x]=d[rs[x]]+1; return x;
}
struct node {int l,r,rt,sz;}stk[N]; int top;
inline void main() {
d[0]=-1;
n=g();
for(R i=1;i<=n;++i) vl[i]=g()-i;
for(R i=1;i<=n;++i) {
++top,stk[top].l=stk[top].r=i,stk[top].rt=i,stk[top].sz=1;
while(top&&vl[stk[top-1].rt]>vl[stk[top].rt]) {
--top;
stk[top].rt=merge(stk[top].rt,stk[top+1].rt);
stk[top].r=stk[top+1].r;
stk[top].sz+=stk[top+1].sz;
while(stk[top].sz>(stk[top].r-stk[top].l+2)/2) {
--stk[top].sz;
stk[top].rt=merge(ls[stk[top].rt],rs[stk[top].rt]);
}
}
}
for(R i=1;i<=top;++i)
for(R j=stk[i].l;j<=stk[i].r;++j)
ans[j]=vl[stk[i].rt],anss+=abs(vl[j]-ans[j]);
printf("%lld\n",anss);
for(R i=1;i<=n;++i) printf("%d ",ans[i]+i);
}
} signed main() {Luitaryi::main(); return 0;}
2020.01.19