AT2289 Yakiniku Restaurants
题目描述
给定你n家之间有距离的烧烤店,一共m个优惠劵,在相应的烧烤店会得到相应的美味度,最后求出把劵全部用完且减去相应的距离的最大值。
算法分析
先说暴力,每次枚举一个起点和一个终点,因为距离一定,所以只用求区间的最大值就行了。复杂度为n2m,只能对30%的点。
再说正确思路,首先观察一下d[i][j]会在哪些答案里求呢,一定是x<j且d[i][x]>d[i][x],y>j且d[i][y]>d[i][x]
那最后d[i][j],最后加到答案里一定是起点为(x,j],终点为[j,y)中的任意组合的结果中,因为是单调的,所以用单调栈维护一下就行了。但显然最后复杂度一定是n2m,能不能用一下单点修改省掉枚举起点和终点的过程呢,显然是可以的。用一下矩阵差分就行了。
注:注意数组大小,和是否开long long
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=5500;
const int M=220;
int n,m,d[N][M],stack[N],L,R,l[N][M],r[N][M],num[N];
long long sq[N][N],ans,a[N];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)scanf("%lld",&a[i]);
for(int i=1;i<n;i++)a[i]+=a[i-1];
for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&d[i][j]);
for(int j=1;j<=m;j++){
L=0;R=0;
for(int i=1;i<=n;i++){
while(L&&stack[L]<d[i][j])L--;
l[i][j]=L?num[L]+1:1;
++L;
stack[L]=d[i][j];
num[L]=i;
}
for(int i=n;i>=1;i--){
while(R&&stack[R]<d[i][j])R--;
r[i][j]=R?num[R]-1:n;
++R;
stack[R]=d[i][j];
num[R]=i;
}
for(int i=1;i<=n;i++){
sq[l[i][j]][i]+=d[i][j];
sq[l[i][j]][r[i][j]+1]-=d[i][j];
sq[i+1][i]-=d[i][j];
sq[i+1][r[i][j]+1]+=d[i][j];
}
}
for(int i=1;i<=n;i++){
for(int j=i;j<=n;j++)sq[i][j]+=sq[i][j-1];
for(int j=i;j<=n;j++)sq[i][j]+=sq[i-1][j];
for(int j=i;j<=n;j++)ans=max(ans,sq[i][j]-a[j-1]+a[i-1]);
}
printf("%lld",ans);
return 0;
}