牛客_ 最小面积子矩阵

题目描述

一个N*M的矩阵，找出这个矩阵中所有元素的和不小于K的面积最小的子矩阵（矩阵中元素个数为矩阵面积）

输入描述:

每个案例第一行三个正整数N,M<=100，表示矩阵大小，和一个整数K

接下来N行，每行M个数，表示矩阵每个元素的值

输出描述:

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

示例1

输入

4 4 10

1 2 3 4

5 6 7 8

9 10 11 12

13 14 15 16

输出

1

Ps:出错时，牛客的用例 只提示一个 99 99 10000   的用例 ，数量很大，无法用来检验。

故此特地提供一个用例，可以供大家简单自己代码使用

输入

4 4 15

0 -2 -7 0

9 2 -6 2

-4 1 -4 1

-1 8 0 -1

输出

6

 

博主的代码：

#include <iostream>
using namespace std;
int main()
{
    int N,M,K;
    while(cin>>N>>M>>K){
        int matrix[N][M];
        for(int i=0;i<N;i++)
            for(int j=0;j<M;j++)
                cin>>matrix[i][j];
        int minS=123123123;
        for(int i=0;i<N;i++){
            int b[M]={0};
            for(int j=i;j<N;j++){
                int linecount=j-i+1;
                for(int k=0;k<M;k++) b[k]+=matrix[j][k];
                int rowcount=0,k=0,temp=0;
                for(;k<M;k++){
                    temp+=b[k];    //temp为 从i~j行，rowcount~k列的总值
                    if(temp>=K){    //该子矩阵达到要求
                        if(minS>linecount*(k-rowcount+1)) //更新最小面积
                            minS=linecount*(k-rowcount+1);
                        if(minS==1){  //已达到所能达到最小值，剪枝
                            printf("1\n");
                            return 0;
                        }
                        k=rowcount++;  //从下一列开始从新计算
                        temp=0;
                    }
                }
            }
        }
        if(minS==123123123)  printf("-1\n");
        else printf("%d\n",minS);
    }
    return 0;
}