上海交通大学【最小面积子矩阵】

题目描述

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

输入描述

每个案例第一行三个正整数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

解题思路

这道题用遍历的方式来实现，复杂度是n*n*m*m。但是也可以降到n*n*m。

我们先通过一个for循环将i从[0-n）赋值，然后保持i不变，再开一个for循环，让j从[i-n）赋值，i和j代表的是第i-j行组成的矩阵。然后再通过一个for循环，将t从[0,m)赋值，代表究竟要有多少列。

我们在这里用b数组来存放从i到j行每一列的值。即b[0]代表的是在第i到j行里0列的值之和。

假设图中红框部分元素之和已经大于k，那么就不必再加入下一列的值。我们可以尝试减去0列的值，看是否大于k，如果还大于k就继续减去左边那一列的和。

以下函数就是用来求出第i到j行所要的最小的列数。

int getMinCol(int* b,int M,int k){
    int sum=0;
    int left=0,right=0;
    int res=-1;
    int tmp=0;
    for(right=0;right<M;right++){
        sum+=b[right];
        while(sum>k){
            tmp=right-left+1;//列数
            if(res==-1 || tmp<res){
                res=tmp;
            }
            sum-=b[left];
            left++;
        }
    }
    return res;
}

最后最小的列数变为2.

那么面积就是(j-i+1)*getMinCol

源码如下：

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

int n,m,k;
int a[105][105];//存放输入
int b[105];//存放列的值，b[0]代表第0列的数字之和

int getMinCol(int* b,int M,int k){
    int sum=0;
    int left=0,right=0;
    int res=-1;
    int tmp=0;
    for(right=0;right<M;right++){
        sum+=b[right];
        while(sum>k){
            tmp=right-left+1;//列数
            if(res==-1 || tmp<res){
                res=tmp;
            }
            sum-=b[left];
            left++;
        }
    }
    return res;
}

int main(){
    while(scanf("%d%d%d",&n,&m,&k)!=EOF){
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                scanf("%d",&a[i][j]);
            }
        }
        int res=-1;
        for(int i=0;i<n;i++){
            memset(b,0,sizeof(b));
            for(int j=i;j<n;j++){
                for(int t=0;t<m;t++){
                    b[t]+=a[j][t];
                }
                int ans=getMinCol(b,m,k)*(j-i+1);//列*行
                if(ans<0) continue;
                if(res==-1 || ans<res)
                    res=ans;
            }
        }
        cout<<res<<endl;

    }
    return 0;
}