最大全1子矩阵

题目:http://ac.jobdu.com/problem.php?cid=1045&pid=0

题目描述:

在一个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
样例输出:
0
4

方法是:

1、先将0/1矩阵读入x,对每一个非零元素x[i][j],将其更新为:在本行,它前面的连续的1的个数+1(+1表示算入自身)

  比如,若某一行为0 1 1 0 1 1 1,则更新为0 1 2 0 1 2 3

2、对每一个非零元素x[i][j],在第j列向上和向下扫描,直到遇到比自身小的数,若扫描了y行,则得到一个大小为x[i][j]*(y+1)的全1子矩阵(+1表示算入自身所在行)

  比如,若某一列为[0 3 4 3 5 2 1]'(方便起见,这里将列表示成一个列向量),我们处理这一列的第4个元素,也就是3,它向上可以扫描2个元素,向下可以扫描1个元素,于是得到一个4×3的全1子矩阵。

3、在这些数值中取一个最大的。

思想大概如下图所示(空白处的0没有标出)

对照步骤2中给出的例子,蓝色的箭头表示向上向下扫描,黑色的框表示最终得到的全1子矩阵

这样做为什么是对的?

想一想,对那个最大的全1子矩阵,用这种方法能不能找到它呢?——肯定可以。

一个最大全1子矩阵,肯定是四个边界中的每一个都不能再扩展了,如下图

假设图中全1子矩阵就是最大子矩阵,则左边界左侧那一列肯定有一个或多个0(否则就可以向左边扩展一列,得到一个更大的全1矩阵)

对其他3个边界有类似的情况。

然后看图中用黑圈标出的1(其特点是:和左边界左侧的某个0在同一行),从这个1出发,按照之前的方法,向上向下扫描,就可以得到这个子矩阵。所以,肯定可以找到。

下面是我的代码,实际实现的时候,为了提高效率,估计了一下upperbound,这个upperbound就是:在当前列,

包含x[i][col]的连续的非零序列的和,比如对某列[0 3 4 3 5 2 1]',后面6个的upperbound都是
3 + 4 + 3 + 5 + 2 + 1 = 18,对于0元素,不需要upperbound

#include <stdio.h>
int m, n;
int x[1002][1002];
int upperbound[1002][1002];
//pre处理后,x[i][j]表示原矩阵第i行中x[i][j]前面有多少个连续的1
void pre() {
    for (int i = 0; i < m; i++) {
        for (int j = 1; j < n; j++) {//注意,j从1开始
            //每行第一个元素不用判断,0/1都不用改变,对应每一段的第一个1也是如此
            if (x[i][j] == 1 && x[i][j - 1] != 0) {
                x[i][j] = x[i][j - 1] + 1;
            }
        }
    }
}
//proc_col对第col列计算每个x[i][col]的upperbound,这个upperbound就是:在当前列,
//包含x[i][col]的连续的非零序列的和,比如对[0 3 4 3 5 2 1],后面6个的upperbound都是
//3 + 4 + 3 + 5 + 2 + 1 = 18,对于0元素,不需要upperbound
void proc_col(int col) {
    for (int i = 0; i < m; i++) {
        if (x[i][col] == 0) {
            continue;
        }
        int sum = 0, j = i;
        while (j < m && x[j][col] > 0) {
            sum += x[j][col];
            j++;
        }
        for (int k = i; k < j; k++) {
            upperbound[k][col] = sum;
        }
        i = j;//之后i还会++,但是不会跳过重要的值,因为此时x[j][]=0或在界外
    }
}
//逐列计算upperbound
void calc_upper(){
    for (int col = 0; col < n; col++) {
        proc_col(col);
    }
}
//从x[row][col]向上向下扫描
int search_up_down(int row, int col) {
    int cnt = 1, val = x[row][col];
    for (int i = row - 1; i >= 0; i--) {
        if (x[i][col] >= val) {
            cnt++;
        } else {
            break;
        }
    }
    for (int i = row + 1; i < m; i++) {
        if (x[i][col] >= val) {
            cnt++;
        } else {
            break;
        }
    }
    return cnt * val;
}
//得到最大全1子矩阵的大小
int getMax() {
    int max = 0;
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (x[i][j] != 0 && max < upperbound[i][j]) {
                int val = search_up_down(i, j);
                if (val > max) {
                    max = val;
                }
            }
        }
    }
    return max;
}
int main(int argc, const char *argv[])
{
    while(scanf("%d%d", &m, &n) != EOF) {
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                scanf("%d", &x[i][j]);
            }
        }
        pre();
        calc_upper();
        printf("%d\n", getMax());
    }
    return 0;
}

原文地址:https://www.cnblogs.com/fstang/p/3087746.html