[蓝桥杯]C++B组 统计子矩阵

有帮助就点个赞呗 👍

暴力方法

思路

• 直接利用二位前缀和计算区域内的sum
• 对于所有满足条件的区域都进行遍历，如长宽分别为（1*1，1*2…）

#include<bits/stdc++.h> 
using namespace std;
const int N=2000;
int n,m,k;
int a[N][N];
/*
3 4 10
2 2 3 4
5 6 7 8
9 10 11 12
*/
int main()
{
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&a[i][j]);
		} 	
	}
	
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
		} 	
		//cout<<endl;
	}
	//cout<<endl;
//	for(int i=1;i<=n;i++)
//	{
//		for(int j=1;j<=m;j++)
//		{
//			cout<<a[i][j]<<" ";
//		} 	
//		cout<<endl;
//	}
	
	int res=0;
	
	for(int x=1;x<=n;x++)//枚举xy，就算左上角和右下角的区域的合
	{
		for(int y=1;y<=m;y++)
		{
			for(int i=1;i<=n;i++)//当区域固定时，求所有的方案
			{
				for(int j=1;j<=m;j++)
				{
					if(i+x-1>n||j+y-1>m) continue;//超出区域
					int sum1=a[i+x-1][j+y-1]-a[x+i-1][j-1]-a[i-1][j+y-1]+a[i-1][j-1];
					if(sum1<=k)
					{
						//cout<<x<<" "<<y<<endl;
						res++; 
					}
						
				} 	
				
			}
		} 	
		
	}
	printf("%d",res);
	
	return 0;
} 

双指针优化(y总方法)

思路

• 因为暴力的方法时间复杂度为O(n^4),会超时，所以必须少一次循环
• 可以用双指针的方法，对于确定的上下边界，里面的每一个元素进行遍历
  

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 510;

int n, m, K;
int s[N][N];

int main()
{
    scanf("%d%d%d", &n, &m, &K);
    
    //读入数据
    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];//注意这里是数据，对y轴不求和
        }
    
    
    LL res = 0;
    for (int i = 1; i <= n; i ++ )//上边界i
        for (int j = i; j <= n; j ++ )//下边界j
            for (int l = 1, r = 1, sum = 0; r <= m; r ++ )//左边界l和右边界r
            {
                sum += s[j][r] - s[i - 1][r];//右边界加一，sum得加上该元素
                //有边界增加了之后，左边界也得看着往右加，不然sum会大于k
                while (sum > K)
                {
                    sum -= s[j][l] - s[i - 1][l];
                    l ++ ;
                }
                res += r - l + 1;//则区间内的元素个数，就是符合条件的区间个数
                //例如上下区间都固定了(i=1,j=3)，左右区间也固定了(l=2,j=5)，那满足条件的区间就是4个
                //分别是 右上角为`（1,5）`，左下角为`(3,2)`,`(3,2)`,`(4,2)`,`(5,2)`,下面有图
            }

    printf("%lld\n", res);
    return 0;
}