第十三届蓝桥杯省赛B组 第六题(前缀和+双指针)

题目
题意: 给定n*m的矩阵，求有多少个子矩阵满足矩阵内和<=k。(n,m<=500)
思路: 降维。枚举列l到r，然后把[l,r]列的数的和看作一个数，那么只看行就具有了单调性。问题转变为在一个一维数组中求有多少个连续子数组，和<=k。用双指针维护即可。
时间复杂度: O(n * m * m)
代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 502;
typedef long long ll;
typedef pair<int,int> PII;
int a[N][N];
int s[N][N]; //第i行的前缀和
int n,m,k,T;
int s2[N]; //把列从l到r看作一个数，维护n行的前缀和
void solve()
{
	cin>>n>>m>>k;
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
		{
			cin>>a[i][j];
			s[i][j] = s[i][j-1] + a[i][j];
		}
	}
	ll ans = 0;
	for(int i=1;i<=m;++i)
	{
		for(int j=i;j<=m;++j)
		{
			for(int t=1;t<=n;++t) 
			{
				s2[t] = s2[t-1] + s[t][j] - s[t][i-1];
			}
			int l = 1;
		    for(int r=1;r<=n;++r) //枚举双指针的右端点
		    {
		    	while(l<=r&&s2[r]-s2[l-1]>k) l++;
		    	ans += r-l+1;
		    }
		}
	}
	cout<<ans;
}
int main(void)
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	solve();
	return 0;
}