4405.统计子矩阵
给定一个 N×M 的矩阵 A,请你统计有多少个子矩阵 (最小 1×1,最大 N×M) 满足子矩阵中所有数的和不超过给定的整数 K?
输入格式
第一行包含三个整数 N,M 和 K。
之后 N 行每行包含 M 个整数,代表矩阵 A。
输出格式
一个整数代表答案。
数据范围
对于 30% 的数据,N,M≤20,
对于 70% 的数据,N,M≤100,
对于 100% 的数据,1≤N,M≤500;0≤Aij≤1000;1≤K≤2.5×108。
输入样例:
3 4 10
1 2 3 4
5 6 7 8
9 10 11 12
输出样例:
19
样例解释
满足条件的子矩阵一共有 19,包含:
大小为 1×1 的有 10 个。
大小为 1×2 的有 3 个。
大小为 1×3 的有 2 个。
大小为 1×4 的有 1 个。
大小为 2×1 的有 3 个。
c++一般有一个时间限制,限制为1s时,计算量在1个亿以内就不会超时
按照暴力做法的话需要四维,时间复杂度为O(n4),我们可以先枚举上下边界,然后就会变成一个一维问题,然后求有多少个区间的总和不超过k
由题意可知矩阵A中所有数均大于等于0,故当右边界确定后(即图中R),L指针越往左,矩阵的和越大,即单调,那么我们就可以用双指针或二分,如果用二分的话,时间复杂度为o(n3logn),感觉也会超时,所以我们先来试一下双指针,如果可以用双指针的话,那就表示可以降到O(n3)
双指针的话要看一下两个指针是不是单调的,即R往右走的时候,L是不是单调往右走,那么我们的L需要用R来定义,所以L一般是R的一个函数,L=f(R),定义完这个函数之后,我们再来证明这个函数是单调的,如果单调的话,我们才可以使用双指针
注:双指针的单调和二分的单调不同,双指针的单调是一个指针随另外一个指针单调,二分的单调指数值随某一个数单调
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 510;
int s[N][N]; //尽管是二维数组,实际上表示一维前缀和
int m,n,k;
int main()
{
scanf("%d%d%d",&n,&m,&k);
// 求解一维前缀和。s[][j]表示第j列的各元素之和
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
{
scanf("%d",&s[i][j]);
s[i][j]+=s[i-1][j];
}
long long res=0;
for (int i=1;i<=n;i++) //枚举第一个横坐标x1范围
for (int j=i;j<=n;j++) // 枚举第二个横坐标x2范围
//这里定下了上下的范围,就该枚举左右的范围了
for (int l=1,r=1,sum=0;r<=m;r++) //双指针,r为快指针
{
// 求解从l到r范围内,每一列的前缀和之和
sum+=s[j][r]-s[i-1][r];
while (sum>k) //此时,应该让l向r的方向靠近,同时sum减去之前l位置的和
{
sum-=s[j][l]-s[i-1][l]; //减去l所在列的和
l++;
}
res+=r-l+1; //l~r范围的子矩阵都符合sum<k
}
printf("%lld\n",res);
return 0;
}