【Luogu5294】[HNOI2019]序列
阅读原文时间:2023年07月10日阅读:1

题目链接

题意

给定一个序列,要求将它改造成一个非降序列,修改一个数的代价为其改变量的平方。

最小化总代价。

另有\(Q\) 次询问,每次修改一个位置上的数。(询问之间独立,互不影响)

Sol

神仙 保序回归 问题,完全不会。

首先是一个暴力的每次 \(O(n)\) 做法。

结论是: 最后的结果序列一定是一段段的相同的数,其值为段中所有元素的平均数。

所以暴力就是维护一个单调栈。

每次加入一个数后形成一段。

然后不断比较栈顶的段和下面一个段的平均数的大小,如果栈顶小一些就把它和下面那个段合并。

然后考虑多组询问。

显然每次重新计算所有的数太呆了,有很多重复计算且没有必要的地方。

一个很直观的想法就是考虑求出最后修改的数所在段的左右端点,这样我们维护一个前缀后缀的答案后就能够 \(O(1)\) 算出最后的答案了。

所以我们先对前后缀分别维护好答案以及单调栈(用可持久化线段树)。

发现如果我们确定了左端点那么右端点是唯一确定的。

首先我们要知道从后往前做上面的贪心也是正确的

然后你现在从后往前已经求出了了 \([R+1,n]\) 这些数的单调栈。

当前段的右端点是 \(R\),然后计算其左端点。

考虑左边来了一个段,其平均数为 \(x\) ,当前平均数为 \(p\)

  1. \(x>p\)

    这时我们显然要把前面那个段给合并进来,那么当前的平均数就会变大。

    由于 \(x\) 往前是单调不升的,所以肯定会在某一个地方停止合并。

  2. \(x<p\)

    那么这个时候已经停止合并,只可能在后面某处停止合并。

所以左端点具有可二分性,可以线段树上二分求出。

然后考虑求右端点。

这个同样具有可二分性,因为当前段合法的话往后再并入一个后的平均值小于后面那一个段,就更加小于并入后的后继段了。

二分套二分求出左右端点就能算答案了。

code:

/*
  1. 一个点最后会在的右端点一定是某个后缀单调栈节点的边界?  通过从后往前暴力的正确性可知
  2. 二分的依据&正确性?  非情况讨论出往前加入一系列单调栈节点后平均值的变化,为一个单峰函数

 */
#include<bits/stdc++.h>
using namespace std;
template<class T>inline void init(T&x){
    x=0;char ch=getchar();bool t=0;
    for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') t=1;
    for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+(ch-48);
    if(t) x=-x;return;
}typedef long long ll;
typedef double db;
const int N=1e5+10;
const int MAXN=N*80;
const int mod=998244353;
template<class T>inline void Inc(T&x,int y){x+=y;if(x>=mod) x-=mod;}
template<class T>inline void Dec(T&x,int y){x-=y;if(x < 0 ) x+=mod;}
template<class T>inline int fpow(int x,T k){int ret=1;for(;k;k>>=1,x=(ll)x*x%mod) if(k&1) ret=(ll)ret*x%mod;return ret;}
inline int Sum(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int Dif(int x,int y){x-=y;if(x < 0 ) x+=mod;return x;}
int maxnum;int n,m;
int A[N];int ans=0,inv[N];
struct node{
    int len;ll sum;int sump;
    node(int _l=0,ll _s=0,int _sump=0){len=_l,sum=_s,sump=_sump;}
    inline int Calc(){int ms=sum%mod;return Dif(sump,(ll)ms*ms%mod*inv[len]%mod);}
    inline node operator +(node b){return node(len+b.len,sum+b.sum,Sum(sump,b.sump));}
    inline node operator -(node b){return node(len-b.len,sum-b.sum,Dif(sump,b.sump));}
    inline bool operator <(node b)const{
        if(!len) return b.len;if(!b.len) return 0;
        return (db)sum/len<(db)b.sum/b.len;
    }
}Su[N];
inline node Get(int l,int r){if(!l&&!r) return node();return Su[r]-Su[l-1];}
struct seg{
    int ls,rs,L,R,Rls;
    inline void Merge(seg LS,seg RS){L=LS.L,R=max(RS.R,LS.R);Rls=LS.Rls;}
}T[MAXN];int cnt=0;
int prelen[N],suflen[N];
int preans[N],sufans[N];
int rtpre[N],rtsuf[N];
node stk[N];int top=0;
inline node Pack(int i){return node(1,A[i],(ll)A[i]*A[i]%mod);}
void Modify(int&u,int l,int r,int p,int ML,int MR){
    T[++cnt]=T[u];u=cnt;
    if(l==r) {T[u].ls=T[u].rs=0,T[u].L=ML,T[u].R=MR;T[u].Rls=MR;return;}
    int mid=(l+r)>>1;
    if(mid>=p) Modify(T[u].ls,l,mid,p,ML,MR);
    else       Modify(T[u].rs,mid+1,r,p,ML,MR);
    T[u].Merge(T[T[u].ls],T[T[u].rs]);
    return;
}
inline int Query_Suf(int u,int l,int r,int p){
    if(l==r) return T[u].R;int mid=(l+r)>>1;
    if(mid>=p) return Query_Suf(T[u].ls,l,mid,p);return Query_Suf(T[u].rs,mid+1,r,p);
}
inline int Query_Pre(int u,int l,int r,int p,node&Re){
    if(r<=p) {
        node tw=Get(T[u].Rls+1,T[u].R),tl=Get(T[u].L,T[u].Rls);
        if(!(tl<(Re+tw))) {Re=Re+Get(T[u].L,T[u].R);return 0;}
        if(l==r) return T[u].R;
    }int mid=(l+r)>>1,pos=0;
    if(p>mid) pos=Query_Pre(T[u].rs,mid+1,r,p,Re);
    if(pos) return pos;return Query_Pre(T[u].ls,l,mid,p,Re);
}
int main()
{
    freopen("sequence.in","r",stdin);
    freopen("sequence.out","w",stdout);
    init(n),init(m);for(int i=1;i<=n;++i) {init(A[i]);Su[i]=Su[i-1]+Pack(i);}
    inv[1]=1;for(int i=2;i<=n;++i) inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod;top=0;
    for(int i=1;i<=n;++i) {
        rtpre[i]=rtpre[i-1];stk[++top]=Pack(i);
        while(top>1&&stk[top]<stk[top-1]) stk[top-1]=stk[top]+stk[top-1],Modify(rtpre[i],1,n,top,0,0),--top;
        Modify(rtpre[i],1,n,top,i-stk[top].len+1,i);
        preans[i]=Sum(preans[i-stk[top].len],stk[top].Calc());
        prelen[i]=top;
    }top=0;
    for(int i=n;i;--i) {
        rtsuf[i]=rtsuf[i+1];stk[++top]=Pack(i);
        while(top>1&&stk[top-1]<stk[top]) stk[top-1]=stk[top]+stk[top-1],Modify(rtsuf[i],1,n,top,0,0),--top;
        Modify(rtsuf[i],1,n,top,i,i+stk[top].len-1);
        sufans[i]=Sum(sufans[i+stk[top].len],stk[top].Calc());
        suflen[i]=top;
    }printf("%d\n",preans[n]);
    while(m--){
        int x,y;init(x),init(y);int dat=A[x];A[x]=y;
        int l=0,r=suflen[x+1]-1,pos=r+1;
        while(l<=r){
            int mid=(l+r)>>1,rp=mid? Query_Suf(rtsuf[x+1],1,n,suflen[x+1]-mid+1):x;
            node val=Pack(x)+Get(x+1,rp);
            int lp=x>1? Query_Pre(rtpre[x-1],1,n,prelen[x-1],val):1;
            if(val<Get(rp+1,Query_Suf(rtsuf[x+1],1,n,suflen[x+1]-mid))) pos=mid,r=mid-1;
            else l=mid+1;
        }
        int rp=pos? Query_Suf(rtsuf[x+1],1,n,suflen[x+1]-pos+1):x;
        node val=Pack(x)+Get(x+1,rp);
        int lp=x>1? Query_Pre(rtpre[x-1],1,n,prelen[x-1],val):1;
        printf("%d\n",Sum(val.Calc(),Sum(preans[lp],sufans[rp+1])));
        A[x]=dat;
    }return 0;
}