九度OJ 1102 最小面积子矩阵

        题目链接：点击打开链接

       

     输入：

      每个案例第一行三个正整数N,M<=100，表示矩阵大小，和一个整数K
      接下来N行，每行M个数，表示矩阵每个元素的值

     输出：

      输出最小面积的值。如果出现任意矩阵的和都小于K，直接输出-1。

     样例输入：

    4 4 10
    1 2 3 4
    5 6 7 8
    9 10 11 12
    13 14 15 16

     样例输出：

    1 

      这条题可以用暴力穷举所有子矩阵的方法AC，时间复杂度是O(n^4)，实际运行时间大约是700多ms。有没有更好的办法呢？

      答案是有的。我们可以选定两列作为边界，将边界内每一行的元素求和，得到一个竖着的一维数组，原问题就转化为求一维数组和不小于S的最短连续子序列。选列边界要用两层外层循环，一边选列一边将行元素求和，时间复杂度为O(n^2)，求最短连续子序列是一个O(n)的算法，总的时间复杂度为O(n^3)，实际运行时间减少到了仅仅20ms。

        

#include <iostream>
#include <string.h>
 
using namespace std;
 
int matrix[100][100], line[100];
int n, m, ans, begin, end;
int goal;
 
int merge(int n) {
    int ans = -1;
    int start = 0, end = 0, sum = 0;
    while (end < n) {
        if (sum < goal)
            sum += line[end];
        while (sum >= goal) {
            int len = end - start + 1;
            if (ans == -1) ans = len;
            else if (len < ans) ans = len;
            sum -= line[start++];
        }
        ++end;
    }
    return ans;
}
 
int main(){
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    while (cin >> n >> m >> goal) {
        ans = -1;
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < m; ++j)
                cin >> matrix[i][j];
        for (int i = 0; i < m; ++i) {
            memset(line, 0, sizeof(line));
            for (int j = i; j < m; ++j) {
                for (int k = 0; k < n; ++k) 
                    line[k] += matrix[k][j];
                int temp = merge(n);
                if (temp != -1) {
                    temp = temp * (j - i + 1);
                    if (ans == -1) ans = temp;
                    else if (temp < ans) ans = temp;
                }
            }
        }
        cout << ans << endl;
    }
    return 0;
}
/**************************************************************
    Problem: 1102
    User: mmclkv
    Language: C++
    Result: Accepted
    Time:20 ms
    Memory:1560 kb
****************************************************************/