AtCoder Regular Contest 067 F - Yakiniku Restaurants 乱搞

题意

有n家饭店和m张餐劵，第i家饭店和第i+1家饭店的距离为Ai，在第i家饭店用第j张餐劵可以得到w[i,j]的美味度。现在要用完这m张餐劵，问最大的（美味度之和）-（走过的距离）是多少。
n<=5000,m<=200,w[i,j],Ai<=10^9

分析

不难发现我们经过的饭店一定是一段连续的区间。那么每张餐劵肯定会被用在这段区间中用这张餐劵收益最大的地方。
一开始的想法是枚举区间左端点和右端点，然后找到每张餐劵的收益，用rmq优化后的复杂度是O(n^2m)，显然不能接受。
这样做的瓶颈在于对于一张餐劵，可能我们在[l,r和[l,r+1]中它带来的收益是一样的，但是我们仍然需要重复计算。
那就考虑如何优化这一步。
一张餐劵什么时候会被用到呢？设lef[i]表示满足$j<i$的最大的j使得在第j家饭店用该餐劵比在第i家饭店用要优秀，rig[i]同理。那么当区间的左端点在[lef[i]+1,i]，右端点在[i,rig[i]-1]时，这张餐劵必然会选择在第i家饭店使用。
如果我们把区间[l,r]看作二维平面上的一个点(l,r)，那么问题就变成了每次给一个子矩形加上一个数，最后询问每一个点的最大值。
直接离线差分即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=5005;
const int M=205;

int n,m,w[N][M],lef[N],rig[N],tot;
LL dis[N],tmp[N],s[N];
struct data{int x,l,r,c;}op[N*M*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

bool cmp(data a,data b)
{
    return a.x<b.x;
}

int main()
{
    n=read();m=read();
    for (int i=2;i<=n;i++) dis[i]=dis[i-1]+(LL)read();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            w[i][j]=read();
    for (int j=1;j<=m;j++)
    {
        for (int i=1;i<=n;i++)
        {
            lef[i]=i-1;
            while (lef[i]&&w[lef[i]][j]<w[i][j]) lef[i]=lef[lef[i]];
        }
        for (int i=n;i>=1;i--)
        {
            rig[i]=i+1;
            while (rig[i]<=n&&w[rig[i]][j]<=w[i][j]) rig[i]=rig[rig[i]];
        }
        for (int i=1;i<=n;i++)
        {
            op[++tot].x=lef[i]+1;op[tot].l=i;op[tot].r=rig[i]-1;op[tot].c=w[i][j];
            op[++tot].x=i+1;op[tot].l=i;op[tot].r=rig[i]-1;op[tot].c=-w[i][j];
        }
    }
    sort(op+1,op+tot+1,cmp);
    LL ans=0;int now=1;
    for (int i=1;i<=n;i++)
    {
        memset(tmp,0,sizeof(tmp));
        while (now<=tot&&op[now].x==i) tmp[op[now].l]+=op[now].c,tmp[op[now].r+1]-=op[now].c,now++;
        for (int j=1;j<=n;j++) tmp[j]+=tmp[j-1],s[j]+=tmp[j],ans=j<i?ans:max(ans,s[j]-dis[j]+dis[i]);
    }
    printf("%lld",ans);
    return 0;
}