子矩阵的和(矩阵前缀和)

题目链接:用户登录 - C语言网

在这里可以模拟一下就知道了,

记录每个 (0,0) 到 (i,j)的矩阵和

然后区间子矩阵的和,就减去多余的部分的矩阵和就可以得到了 子矩阵的和

然后 这里最好使用 下标 1 ~ n 到 1 ~ m 存储,这样就可以方便,根据一条规律来使用即可。

初始化函数

cpp 复制代码
// 初始化矩阵前缀和
inline void Init(){
    for(int i = 1;i <= n;++i)
	    for(int j = 1;j <= m;++j)
    	    s[i][j] = s[i - 1][j] + s[i][j - 1] + a[i][j] - s[i - 1][j - 1];
}

获取矩阵和函数

cpp 复制代码
// 根据左上角坐标以及右下角坐标获取矩阵和
inline int getSum(PII p1,PII p2){
	int x1 = p1.x,y1=p1.y;
	int x2 = p2.x,y2=p2.y;
	return s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
}

题解代码

cpp 复制代码
#include <bits/stdc++.h>
#define int long long
#define endl '\n'
#define x first
#define y second
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
using PII = pair<int,int>;
const int N = 510;
int a[N][N],s[N][N],n,m,k,ans;
// 初始化矩阵前缀和
inline void Init(){
    for(int i = 1;i <= n;++i)
	    for(int j = 1;j <= m;++j)
    	    s[i][j] = s[i - 1][j] + s[i][j - 1] + a[i][j] - s[i - 1][j - 1];
}

// 根据左上角坐标以及右下角坐标获取矩阵和
inline int getSum(PII p1,PII p2){
	int x1 = p1.x,y1=p1.y;
	int x2 = p2.x,y2=p2.y;
	return s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
}
signed main(){
	IOS;
	cin >> n;
	m = n;
	for(int i = 1;i <= n;++i)
	    for(int j = 1;j <= m;++j)
			cin >> a[i][j];

// 初始化矩阵前缀和
	Init();
	
// 	循环遍历所有上下坐标,并获取矩阵和
	for(int i = 1;i <= n;++i){
		for(int j = 1;j <= m;++j){
			PII leftUp = PII(i,j);
			for(int x = i;x <= n;++x){
			    for(int y = j;y <= m;++y){
			    	PII rightDown = PII(x,y);
					int sum = getSum(leftUp,rightDown);
					ans = max(sum,ans);
				}
			}
		}
	}

	cout << ans << endl;
	return 0;
}

最后提交

相关推荐
luofeiju7 小时前
使用LU分解求解线性方程组
线性代数·算法
FF-Studio13 小时前
【硬核数学 · LLM篇】3.1 Transformer之心:自注意力机制的线性代数解构《从零构建机器学习、深度学习到LLM的数学认知》
人工智能·pytorch·深度学习·线性代数·机器学习·数学建模·transformer
szekl20 小时前
HDMI 2.0 4×2矩阵切换器412HN——多信号输入输出的高清解决方案
linux·矩阵·计算机外设·电脑·ekl
盛寒2 天前
矩阵的定义和运算 线性代数
线性代数
盛寒2 天前
初等变换 线性代数
线性代数
叶子爱分享2 天前
浅谈「线性代数的本质」 - 系列合集
线性代数
luofeiju2 天前
RGB下的色彩变换:用线性代数解构色彩世界
图像处理·人工智能·opencv·线性代数
好开心啊没烦恼2 天前
Python:线性代数,向量内积谐音记忆。
开发语言·python·线性代数·数据挖掘·数据分析
引量AI11 天前
TikTok 矩阵如何快速涨粉
大数据·人工智能·矩阵·tiktok矩阵·海外社媒
Ven%12 天前
矩阵阶数(线性代数) vs. 张量维度(深度学习):线性代数与深度学习的基石辨析,再也不会被矩阵阶数给混淆了
人工智能·pytorch·深度学习·线性代数·矩阵·tensor·张量