HihoCoder 1502 : 最大子矩阵 (双指针)

描述

给定一个NxM的矩阵A和一个整数K，小Hi希望你能求出其中最大（元素数目最多）的子矩阵，并且该子矩阵中所有元素的和不超过K。

输入

第一行包含三个整数N、M和K。

以下N行每行包含M个整数，表示A。

对于40%的数据，1 <= N, M <= 10

对于100%的数据，1 <= N, M <= 250 1 <= K <= 2147483647 1 <= Aij <= 10000

输出

满足条件最大的子矩阵所包含的元素数目。如果没有子矩阵满足条件，输出-1。

样例输入

3 3 9
1 2 3
2 3 4
3 4 5

样例输出

4 

由于A>=1满足区间和的单调性。所以可以用双指针，即枚举矩形的上下边界，然后移动左右边界。

qwq，开始看“元素”，我以为要去重，所以做复杂了。

（总的来说，很多可以二分做的题，可以用双指针，效率更高。）

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=;
int N,M,K,a[maxn][maxn],sum[maxn][maxn];
int vis[],num,Now,ans=-;
int main()
{
    int i,j,k,L,R;
    scanf("%d%d%d",&N,&M,&K);
    for(i=;i<=N;i++)
      for(j=;j<=M;j++){
        scanf("%d",&a[i][j]);
        sum[i][j]=sum[i-][j]+a[i][j];
    }

    for(i=;i<=N;i++)
      for(j=i;j<=N;j++){
         memset(vis,,sizeof(vis));
         num=; Now=;
         for(R=,L=;R<=M;R++){
             num+=j-i+; Now+=sum[j][R]-sum[i-][R];
             while(Now>K&&L<=R) {
                num-=j-i+;
                Now-=sum[j][L]-sum[i-][L];
                L++;
             }
             if(Now<=K&&L<=R) ans=max(ans,num);
         }
    }
    printf("%d\n",ans);
    return ;
}