P3017 [USACO11MAR]布朗尼切片Brownie Slicing

题目传送

很好的一道二分题。

二分答案+贪心+前缀和优化

对于二分的答案x，我们对于每一条，先假设宽为1，看看能不满足分为不小于x的b块，如果不满足，宽度就加宽，如此处理即可。
一开始循环中的终止条件写错，调了好长时间。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath> 
#define LL long long
using namespace std;
int n,m,A,B,a[509][509],s[509][509];
int b[509];
bool check(int x)
{
    int last=0,i,j,h=1;
    while(last<n&&h<=A)
    {
        int hx=h;
        for(i=1;i<=n-last;i++)
        {
            for(j=1;j<=m;j++)
               b[j]=s[i+last][j]-s[last][j];
            int num=0,sum=0;
            j=1;
            while(num<B&&j<=m)
            {
                j=lower_bound(b+j,b+m+1,sum+x)-b;
                if(j>m) break;
                sum=b[j];
                j++;
                num++;
            }
            if(num<B) continue;
            else
            {
                last+=i;
                h++;
                break;
            }
        }
        if(hx==h) return 0;
    }
    if(h<=A) return 0;
    return 1;
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&A,&B);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=m;j++)
     {
        scanf("%d",&a[i][j]);
        s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]; 
     }

    int L=0,R=1e9+1,mid,ans=R;
    while(L<=R)
    {
        mid=(L+R)>>1;
        if(check(mid)) ans=mid,L=mid+1;
        else R=mid-1;
    }
    printf("%d\n",ans);
    return 0;
}