洛谷模拟赛题解

P9744 「KDOI-06-S」消除序列

首先我们可以想到一个很显然的方法，就是暴力去枚举你要执行操作1 的位置，对于在区间里的位置，我们对他进行操作3，然后不在的我们把1都进行操作2，这就是20pts

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f3f3f3f3f
#define int long long
using namespace std;
const int N=4e4+10;
template<class T> inline T &read(T &x){
    bool f=1;x=0;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) f^=(ch=='-');
    for(;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+(ch^48);
    return f?x:(x=-x);
}
int n,a[N],b[N],c[N],q,m,p[N],vis[N];
signed main() {
    read(n);
    for(int i=1;i<=n;++i) read(a[i]);
    for(int i=1;i<=n;++i) read(b[i]);
    for(int i=1;i<=n;++i) read(c[i]);
    read(q);
    while(q--){
        read(m);
        int ans=INF;
        for(int i=1;i<=n;++i) p[i]=0;
        for(int i=1;i<=m;++i) read(p[i]);
        if(m==n){
            puts("0");
            continue;
        }
        for(int i=1;i<=n;++i){//枚举从哪一位施加操作1 
            int cnt=a[i],j=0;
            for(int i=1;i<=n;++i) vis[i]=0;
            for(j=1;j<=m&&p[j]<=i;++j) cnt+=c[p[j]];//操作1中区间中变1的代价
            for(;j<=m;++j) vis[p[j]]=1;//p[]存的是哪一位最后必须是1，也就是说操作1以外的区间只能变0
            for(int k=i+1;k<=n;++k) if(!vis[k]) cnt+=b[k];
            ans=min(ans,cnt);
        }
        printf("%lld\n",ans);
    }
    return 0;
}

显然，这个很丑陋的代码还能继续优化，我们发现每次用 $O(n)$ 枚举操作1的右端点是非常浪费的，因为我们可以用一个很简单的递推来求出每个 $i$ 位置我们把它都变成 $0$ 的最小代价，这就是一个小优化

但是，时间复杂度还是不够优秀，我们很容易转换到 $p[]$ 数组本身去考虑把最终区间分割的1们

设
$f_i$ 为把 $1\sim p_i$ 全为1改变到满足题意的最小答案

$g_i$ 为把序列全为0变为满足题意的最小答案

对于一个位置 $p_i$ 我们有两种转移的方法，分别是
① $f_{i-1}+\sum _{j=p_{i-1}+1}^{p_i-1}{b}_i$ 其中 $\sum _{j=p_{i-1}+1}^{p_i-1}{b}_i$ 可以用前缀和来维护

② 把 $1\sim p_i$ 全变为0，然后再进行更改，$a_{p_i-1}+g_{i-1}$

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e6+10;
template<class T>inline T &read(T &x){
    bool f=1;x=0;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) f^=(ch=='-');
    for(;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+(ch^48);
    return f?x:(x=-x);
}
int n,q,m,a[N],b[N],c[N],p[N];
ll s[N],g[N],f[N];
int main() {
    read(n);
    for(int i=1; i<=n; i++) read(a[i]);
    for(int i=1; i<=n; i++)
        read(b[i]),s[i]=s[i-1]+b[i];
    for(int i=1; i<=n; i++) read(c[i]);
    for(int i=1; i<=n; i++)
        a[i]=min(a[i],a[i-1]+b[i]);
    read(q);
    while(q--) {
        read(m);
        for(int i=1; i<=m; i++) read(p[i]);
        for(int i=1; i<=m; i++) {
            g[i]=g[i-1]+c[p[i]];
            f[i]=a[p[i]-1]+g[i-1];
            f[i]=min(f[i],f[i-1]+s[p[i]-1]-s[p[i-1]]);
        }
        printf("%lld\n",min(g[m]+a[n],f[m]+s[n]-s[p[m]]));
    }
    return 0;
}