Loj #3059. 「HNOI2019」序列

Loj #3059. 「HNOI2019」序列

给定一个长度为 \(n\) 的序列 \(A_1, \ldots , A_n\),以及 \(m\) 个操作,每个操作将一个 \(A_i\) 修改为 \(k\)。第一次修改之前及每次修改之后,都要求你找到一个同样长度为 \(n\) 的单调不降序列 \(B_1, \ldots , B_n\),使得 \(\sum_{i=1}^n (A_i −B_i)^2\) 最小,并输出该最小值。需要注意的是每次操作的影响都是独立的,也即每次操作只会对当前询问造成影响。为了避免精度问题,我们保证这个最小值是个分数,也即能表示为两个非负整数相除的形式:\(x/y\)。那么你将要输出 \((x\times y^{P-2})\bmod P\) 的值,表示模意义下 \(x/y\) 的值。其中 \(P=998244353\) 是一个大质数。

输入格式

第一行两个非负整数 \(n,m\),代表序列长度和操作数。

第二行有 \(n\) 个由空格隔开的正整数,代表序列 \(A_1, \ldots , A_n\)

接下来 \(m\) 行每行两个正整数 \(i, k\),代表将 \(A_i\) 修改为 \(k\)

输出格式

输出 \(m + 1\) 行每行一个整数,第 \(i\) 行输出第 \(i − 1\) 次修改后的答案。特别的,第 \(1\) 行应为初始局面的答案。

数据范围与提示

对于前 \(10\%\) 的数据,保证 \(n, m \le 10\)\(k, A_i ≤ 1000\),且存在一种最优方案,使得 \(B_i\) 皆为整数。

对于前 \(30\%\) 的数据,保证 \(n, m \le 100\)

对于另外 \(20\%\) 的数据,保证 \(m = 0\)

对于另外 \(20\%\) 的数据,保证 \(n, m \le 3 \times 10^4\)

对于所有数据,保证 \(3 \le n \le 10^5, 0 \le m \le 10^5, 1 \le k, A_i \le 10^9\)

\(\\\)

扫描二维码关注公众号,回复: 5876804 查看本文章

Orz

假设没有修改,那这就是个经典问题(然而我不会)

如果\(A\)也是个单调不下降序列,那么答案就是\(0\)。否则,对于\(A_i>A_{i+1}\)的情况,我们将\(A_i\)\(A_{i+1}\)合并起来。合并后的块之间也存在这种关系,不过是拿块的平均值比较。

\(B_i\)就是\(i\)所在块的平均值。

我们先将询问离线下来,按位置排序。假设当前处理位置\(i\)的询问,我们维护两个栈,一个是从\(1\)\(i-1\),按上述规则合并后的栈;一个数从\(i+1\)\(n\)按上述规则合并后的栈。对于后一个栈,我们可以先从\(n\)\(1\)维护一遍,记录下每个位置\(i\)加入栈中对这个栈的修改。然后依次回退。

假设修改后\(i\)所在的块为\([L,R]\),那么\([L,R]\)\([1,L-1]\)\([R+1,n]\)分别形成的栈之间是相互独立的。这也就是维护两个栈的原因。

考虑求\([L,R]\)。我们先二分出\(R\),然后再找对应的\(L\),判断\([L,R]\)的平均值是否\(\leq\)\(R+1\)所在块的平均值,如果是,将二分边界往左移;否则往右移(注意这里二分的是完整的块)。因为如果以\(R\)所在块为右端点成立,那么我们加入\(R\)所在块右侧的块依然成立,所以答案是有单调性的。找\(R\)对应的\(L\)用的也是相同的原理。

代码:

#include<bits/stdc++.h>
#define ll long long
#define N 100005

using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}

const ll mod=998244353;
ll ksm(ll t,ll x) {
    ll ans=1;
    for(;x;x>>=1,t=t*t%mod)
        if(x&1) ans=ans*t%mod;
    return ans;
}

int n,m;
ll a[N];

struct node {
    ll sum,size;
    int l,r;
    long double key;
    node() {}
    node(ll _sum,ll _size,int _l,int _r) {
        sum=_sum,size=_size;
        l=_l,r=_r;
        key=(long double)sum/size;
    }
};

node operator +(const node &a,const node &b) {return node(a.sum+b.sum,a.size+b.size,min(a.l,b.l),max(a.r,b.r));}

int cal() {
    static node st[N];
    int top;
    st[top=1]=node(a[1],1,1,1);
    node tem;
    for(int i=2;i<=n;i++) {
        tem=node(a[i],1,i,i);
        while(top>=1&&st[top].key>=tem.key) {
            tem=tem+st[top];
            top--;
        }
        st[++top]=tem;
    }
    int now=1;
    ll ans=0;
    for(int i=1;i<=top;i++) {
        ll ave=st[i].sum%mod*ksm(st[i].size,mod-2)%mod;
        for(int j=1;j<=st[i].size;j++,now++) {
            (ans+=(a[now]-ave)*(a[now]-ave))%=mod;
        }
    }
    return ans;
}

int rx[N],lx[N];
ll sum[N];
ll suf[N],pre[N],sums[N];

int t1=0,t2=0;
node pres[N],sufs[N];
int pos[N];
node del[N];

struct query {
    int id,k;
};

vector<query>q[N];
int Find_left(int x,ll val,int R) {
    if(x==1) return 1;
    if(pres[t1].key<=(long double)(sum[R]-sum[x]+val)/(R-x+1)) return x;
    int l=1,r=t1,mid;
    while(l<r) {
        mid=l+r+1>>1;
        int L=pres[mid].l;
        long double k=(long double)(sum[R]-sum[L-1]-a[x]+val)/(R-L+1);
        if(pres[mid-1].key<=k) l=mid;
        else r=mid-1;
    }
    return pres[l].l;
}

int Find_right(int x,ll val) {
    if(x==n) return n;
    int L=Find_left(x,val,x);
    if(sufs[t2].key>=(long double)(sum[x-1]-sum[L-1]+val)/(x-L+1)) return x;
    int l=1,r=t2,mid;
    while(l<r) {
        mid=l+r+1>>1;
        int R=sufs[mid].r;
        int L=Find_left(x,val,R);
        long double k=(long double)(sum[R]-sum[L-1]-a[x]+val)/(R-L+1);
        if(k>sufs[mid-1].key) r=mid-1;
        else l=mid;
    }
    return sufs[l].r;
}

ll Ans[N];

int main() {
    n=Get(),m=Get();
    for(int i=1;i<=n;i++) a[i]=Get();
    
    for(int i=1;i<=n;i++) {
        sum[i]=sum[i-1]+a[i];
        sums[i]=(sums[i-1]+a[i]*a[i])%mod;
    }
    cout<<cal()<<"\n";
    
    for(int i=1;i<=m;i++) {
        int x=Get(),k=Get();
        q[x].push_back((query) {i,k});
    }
    
    node tem;
    for(int i=n;i>=1;i--) {
        tem=node(a[i],1,i,i);
        pos[i]=t2;
        while(t2&&sufs[t2].key<=tem.key) {
            tem=tem+sufs[t2];
            t2--;
        }
        del[i]=sufs[++t2];
        sufs[t2]=tem;
        
        ll ave=tem.sum%mod*ksm(tem.size,mod-2)%mod;
        int R=tem.r;
        suf[i]=((suf[R+1]+sums[R]-sums[i-1]+ave*ave%mod*(R-i+1)-2*ave*((sum[R]-sum[i-1])%mod))%mod+mod)%mod;
    }
    
    pres[0].key=0;
    sufs[0].key=1e9+7;
    for(int i=1;i<=n;i++) {
        sufs[t2]=del[i];
        t2=pos[i];
        for(int j=0;j<q[i].size();j++) {
            int x=i;
            ll k=q[i][j].k;
            ll dlt=(k-a[x]+mod)%mod,dlts=(k*k-a[x]*a[x])%mod+mod;
            int R=Find_right(x,k),L=Find_left(i,k,R);
            ll ans=(pre[L-1]+suf[R+1])%mod;
            ll ave=(sum[R]-sum[L-1]+dlt+mod)%mod*ksm(R-L+1,mod-2)%mod;
            ans=((ans +sums[R]-sums[L-1]+dlts +ave*ave%mod*(R-L+1)%mod -2*ave*((sum[R]-sum[L-1]+dlt)%mod))%mod+mod)%mod;
            Ans[q[i][j].id]=ans;
        }
        tem=node(a[i],1,i,i);
        while(t1&&pres[t1].key>=tem.key) {
            tem=tem+pres[t1];
            t1--;
        }
        pres[++t1]=tem;
        
        ll ave=tem.sum%mod*ksm(tem.size,mod-2)%mod;
        int L=tem.l;
        pre[i]=((pre[L-1]+sums[i]-sums[L-1]+ave*ave%mod*(i-L+1)%mod-2*ave*((sum[i]-sum[L-1])%mod))%mod+mod)%mod;
    }
    for(int i=1;i<=m;i++) cout<<Ans[i]<<"\n";
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hchhch233/p/10702421.html
今日推荐