【单调队列优化】[CF372C] Watching Fireworks is Fun

突然发现我可能单调队列都打不来了...我太菜了...

 

这道题显然有$$f[i][j]=min\{f[i-1][k]+\vert j-a[i] \vert\}$$

 

则$ans=\sum_{i=1}^{m} b_i - min_{j=1}^{n}\{f[m][j]\}$

 

令$len=(t[i]-t[i-1])*d$则其中k满足$$k∈[j-len,j+len]$$

 

 

$f[i][]$只与$f[i-1][]$有关，所以可以把第一维压掉

 

不难弄出一个$O(n^2m)$的代码：

 

#include<bits/stdc++.h>
#define int long long 
#define writeln(x)  write(x),puts("")
#define writep(x)   write(x),putchar(' ')
using namespace std;
inline int read(){
    int ans=0,f=1;char chr=getchar();
    while(!isdigit(chr)){if(chr=='-') f=-1;chr=getchar();}
    while(isdigit(chr)){ans=(ans<<3)+(ans<<1)+chr-48;chr=getchar();}
    return ans*f;
}void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}const int M = 150005;
int f[2][M],n,m,d,ans=0x3f3f3f3f,sum;
struct P{int a,b,t;}a[M];
signed main(){
    n=read(),m=read(),d=read(),sum=0;
    for(int i=1;i<=m;i++){
        int x=read(),y=read(),z=read();
        sum+=y;
        a[i]=(P){x,y,z};
    }
    for(int i(1);i<=n;i++)f[1][i]=abs(a[1].a-i);
    for(int i(2);i<=m;i++){
        memset(f[i&1],0x3f,sizeof(f[i&1]));
        int len=(a[i].t-a[i-1].t)*d;
        for(int j=1;j<=n;j++){
            for(int k=max(1ll,j-len);k<=min(j+len,n);k++)
                f[i&1][j]=min(f[i&1][j],f[i&1^1][k]+abs(a[i].a-j));
            if(i==m)ans=min(ans,f[i&1][j]);
        }
    }
    cout<<sum-ans<<endl;
    return 0;
}

 

但是显然会T，注意到上面k的范围，可以考虑单调队列优化，对于每一次的$i,j$，对$[j-len,j]$跑一次滑动窗口，对$[j,j+len]$跑一次滑动窗口取最小值即可

 

#include<bits/stdc++.h>
#define int long long
#define writeln(x)  write(x),puts("")
#define writep(x)   write(x),putchar(' ')
using namespace std;
inline int read(){
    int ans=0,f=1;char chr=getchar();
    while(!isdigit(chr)){if(chr=='-') f=-1;chr=getchar();}
    while(isdigit(chr)){ans=(ans<<3)+(ans<<1)+chr-48;chr=getchar();}
    return ans*f;
}void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}const int M = 150005;
int f[2][M],n,m,d,ans=0x3f3f3f3f,sum,q[M];
struct P{int a,b,t;}a[M];
signed main(){
    n=read(),m=read(),d=read(),sum=0;
    for(int i=1;i<=m;i++){
        int x=read(),y=read(),z=read();
        sum+=y;a[i]=(P){x,y,z};
    }for(int i(1);i<=n;i++)f[1][i]=abs(a[1].a-i);
    for(int i(2);i<=m;i++){
        int len=(a[i].t-a[i-1].t)*d,l=1,r=0;
        for(int j=1;j<=n;j++){
            while(l<=r&&q[l]<j-len)++l;
            while(l<=r&&f[i&1^1][q[r]]>f[i&1^1][j])--r;
            q[++r]=j;
            f[i&1][j]=f[i&1^1][q[l]]+abs(j-a[i].a);
            if(i==m)ans=min(ans,f[i&1][j]);
        }l=1,r=0;
        for(int j=n;j>=1;j--){
            while(l<=r&&q[l]-len>j)++l;
            while(l<=r&&f[i&1^1][q[r]]>f[i&1^1][j])--r;
            q[++r]=j;
            f[i&1][j]=min(f[i&1][j],f[i&1^1][q[l]]+abs(j-a[i].a));
            if(i==m)ans=min(ans,f[i&1][j]);
        }        
    }cout<<sum-ans<<endl;
    return 0;
}