题目描述
一个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;
}