九度刷题

题目描述：

在一个M * N的矩阵中，所有的元素只有0和1，从这个矩阵中找出一个面积最大的全1子矩阵，所谓最大是指元素1的个数最多。

输入：

输入可能包含多个测试样例。
对于每个测试案例，输入的第一行是两个整数m、n(1<=m、n<=1000)：代表将要输入的矩阵的大小。
矩阵共有m行，每行有n个整数，分别是0或1，相邻两数之间严格用一个空格隔开。

输出：

对应每个测试案例，输出矩阵中面积最大的全1子矩阵的元素个数。

样例输入：

2 2
0 0
0 0
4 4
0 0 0 0
0 1 1 0
0 1 1 0
0 0 0 0

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define MAXN 2010
int h[MAXN],r[MAXN],l[MAXN];
int map[MAXN][MAXN];
char str[2];
int n,m;
void all_1_matrix()
{
    int i,j,max = 0;
    memset(h,0,sizeof(h));
    for(i = 0; i < n; i++)
    {
        for(j = 0 ; j < m; j++)
        {
            if(map[i][j] == 1)
                h[j] ++;
            else h[j] = 0;
        }
        // 类类似并查集的路径压缩
        // left 求高度为h〔j〕的矩阵的左边界
        for(j = 0; j < m; j ++)
        {
            l[j] = j;
            while(l[j] > 0 && h[j] <= h[l[j] - 1])
                l[j] = l[l[j] - 1];
        }
        // right求高度为h〔j〕的矩阵的右边界
        for(j = m-1; j >= 0; j --)
        {
            r[j] = j;
            while(r[j] < m-1 && h[j] <=h[r[j] + 1])
                r[j] = r[r[j] + 1];
        }
        for(j = 0; j < m; j ++)
        {
            if((r[j] - l[j] + 1) * h[j] > max)
                max = (r[j] - l[j] + 1) * h[j];
        }
    }
    printf("%d\n",max);
}
int main()
{
    int t, i , j, k, len ;
    char ch;

    while(scanf("%d %d",&n,&m)!=EOF)
    {
      // scanf("%d %d",&n,&m);
        for(i = 0; i < n; i++)
        {
            for(j = 0; j < m; j++)
            {
               scanf("%d",&map[i][j]);
            }
        }
        all_1_matrix();
    }
}