第十三届蓝桥杯大赛软件赛省赛C/C++ 大学 B 组 统计子矩阵

cpp 复制代码
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<queue>

using namespace std;

int cnt,temp;
int n,m,K;
int a[505][505];
int pre[505][505];//二维前缀和



void sol()
{
	cin>>n>>m>>K;
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
		{
			cin>>a[i][j];
			pre[i][j]=a[i][j]+pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1];
		}
	}
	int limit;//等到第四层循环时,在k行l列时超过K,记录下其l值。当i,j不变,k为k+1行时,再次达到l时一定会再次超过K,所以进行剪枝
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
		{
			limit=m;
			for(int k=i;k<=n;++k)
			{
				for(int l=j;l<=m&&l<=limit;++l)
				{
					temp=pre[k][l]-pre[i-1][l]-pre[k][j-1]+pre[i-1][j-1];
					if(temp<=K)++cnt;
					else {
						limit=l;
						break;
					}
				}
			}
		}
	}
	
    cout<<cnt<<endl;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	sol();
}
相关推荐
梁辰兴1 小时前
数据结构:排序
数据结构·算法·排序算法·c·插入排序·排序·交换排序
Murphy_lx1 小时前
Lambda表达式
开发语言·c++
yangpipi-1 小时前
C++并发编程-23. 线程间切分任务的方法
开发语言·c++
野犬寒鸦1 小时前
力扣hot100:搜索二维矩阵 II(常见误区与高效解法详解)(240)
java·数据结构·算法·leetcode·面试
菜鸟得菜1 小时前
leecode kadane算法 解决数组中子数组的最大和,以及环形数组连续子数组的最大和问题
数据结构·算法·leetcode
楼田莉子2 小时前
C++算法专题学习——分治
数据结构·c++·学习·算法·leetcode·排序算法
一支鱼3 小时前
leetcode常用解题方案总结
前端·算法·leetcode
ulias2123 小时前
各种背包问题简述
数据结构·c++·算法·动态规划
m0_570466413 小时前
代码随想录算法训练营第二十八天 | 买卖股票的最佳实际、跳跃游戏、K次取反后最大化的数组和
java·开发语言·算法
程序喵大人3 小时前
分享个C++线程池的实现源码
开发语言·c++·线程池