【JZOJ 4820】 最大化

Description

Analysis

n^4的方法很好想，枚举答案，判断是否可行。
100分的方法要n^3logn
枚举左右边界，那么我们就可以把中间几列数压成一个数。
问题变成序列上选取连续最长一段使得其和大于0。
我们可以做一个前缀和。
枚举k，对于起点i,j。若$i<j$且$sum[i]<sum[j]$显然j决策时冗余的。
因此我们顺着做，维护一个单调递减的队列，每次只需在队列里面二分出一个最靠左的
使得sum[k]-sum[i]>0的位置即可。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
typedef long long ll;
const int N=310;
int n,m;
ll num,a[N][N],b[N],sum[N],q[N],p[N],f[N][N];
int find(ll x)
{
    int l=0,r=num,mid;
    while(l<r)
    {
        mid=(l+r)>>1;
        if(x-q[mid]>0) r=mid;
        else l=mid+1;
    }
    return p[l];
}
int main()
{
    freopen("max.in","r",stdin);
    freopen("max.out","w",stdout);
    scanf("%d %d",&n,&m);
    fo(i,1,n)
        fo(j,1,m)
        {
            scanf("%lld",&a[i][j]);
            f[i][j]=f[i][j-1]+a[i][j];
        }
    ll ans=0;
    fo(i,1,m)
        fo(j,i,m)
            if((j-i+1)*n>ans)
            {
                fo(k,1,n) b[k]=f[k][j]-f[k][i-1],sum[k]=sum[k-1]+b[k];
                memset(q,0,sizeof(q));
                num=0;
                fo(k,1,n)
                {
                    if(sum[k]<=q[num]) q[++num]=sum[k],p[num]=k;
                    int pos=find(sum[k]);
                    ll size=(j-i+1)*(k-pos);
                    if(size>ans) ans=size;
                }
            }
    printf("%lld",ans);
    return 0;
}