[Jobdu] 題目1497：面積最大的全1子矩陣

題目描述：

在一個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的個數，下面分別利用直方圖的算法來求解每一行中的最大值，然后求出全局最大值。代碼如下：

 

#include <iostream>
#include <stack>
#include <cstdio>
using namespace std;
 
int m, n;
int a[1000][1000];
int b[2][1001];
int f = 0;
 
int getMax() {
    stack<int> s;
    b[f][n] = 0;
    int max = 0, tmp;
    for (int i = 0; i <= n; ++i) {
        if (s.empty() || b[f][s.top()] < b[f][i]) {
            s.push(i);
        } else {
            int idx = s.top();
            s.pop();
            tmp = b[f][idx] * (s.empty() ? i : i - s.top() - 1);
            max = (max > tmp) ? max : tmp;
            --i;
        }
    }
    return max;
}
 
void getRes() {
    int max = getMax(), tmp;
    ++f; f %= 2;
    for (int i = 1; i < m; ++i) {
        for (int j = 0; j < n; ++j) {
            b[f][j] = (a[i][j] == 0) ? 0 : b[(f+1)%2][j] + 1;
        }
        tmp = getMax();
        ++f; f %= 2;
        max = (max > tmp) ? max : tmp;
    }
    cout << max << endl;
}
 
int main() {
    //freopen("1497.in", "r", stdin);
    while (cin >> m >> n) {
        f = 0;
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                cin >> a[i][j];
                if (i == 0)
                    b[0][j] = a[0][j];
            }
        }
        getRes();
    }
    return 0;
}
 
/**************************************************************
    Problem: 1497
    User: hupo250
    Language: C++
    Result: Accepted
    Time:800 ms
    Memory:5432 kb
****************************************************************/