算法基础之二维前缀和

2021/12/2 20:07:12

本文主要是介绍算法基础之二维前缀和,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

更新啦~

二维前缀和

我们一样先从题目来插入
 首先 
输入用例为: 
3 4 2
1 2 3 4
5 6 7 8
9 10 11 12
1 1 1 4
1 1 2 4
表示的是输入3行4列个数并进行2次查询
然后是3行4列个数
最后查询是从第1,1到第1,4的二维数组和,输出和

为了方便我把初始的坐标称为x1,y1  终点坐标x2,y2

并把原数据存入N[3][4]里面,矩阵和存在sum[3][4]

 同样的遇到这种题目 暴力破解可以,直接从1,1到1,4的数组和加起来就行

 

但是如果数列比较大而查询的次数又比较多,别如10个数查询50次

每次遍历1遍,需要500次,就会使得算法复杂度为O(n*m)

所以为了优化算法,降低复杂度,就出现了二维前缀和

二维前缀和的公式

 

 

 所以根据这个方法我们可以得到一个公式(求子矩阵时的)

黄矩阵和=红矩阵和-粉矩阵和-绿矩阵和+黑矩阵和

 子矩阵和=sum[x2][y2]-sum[x1][y1-1]-sum[x1][y1-1]+sum[x1-1][y1-1]

这个矩阵和思想也差不多,只不过是公式的变形

红矩阵和=黄矩阵和+粉矩阵和+绿矩阵和-黑矩阵和

哦注意此时的黄矩阵是一个原数而不是一个矩阵和

红矩阵即矩阵和

 sum[i][j]=N[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]

二维前缀和     算法复杂度O(m)


主要思想:就是遍历存矩阵和,然后查询,运用这个公式即可求出
矩阵和 即可求解 

#include<stdio.h>
int N[100][100],sum[100][100];
void Sum(int n,int m)//首先求红矩阵和
{
		for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		sum[i][j]=N[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
	}	
}


void print(int x1,int y1,int x2,int y2)//求黄矩阵和并输出
{
	printf("%d\n",sum[x2][y2]-sum[x1][y1-1]-sum[x1][y1-1]+sum[x1-1][y1-1]);
}

int main()
{
	int n,m,M;
	scanf("%d %d %d",&n,&m,&M);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		scanf("%d",&N[i][j]);
	}
	Sum(n,m);
	int x1,y1,x2,y2;
	while(M--)
	{
		scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
		print(x1,y1,x2,y2);
	}
	return 0;
}

这篇文章以互相学习为主,有什么错的还望告知,谢谢啦

(ps :小声bb, 如果有人看的话,我就会持续更新算法基础哦)



这篇关于算法基础之二维前缀和的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程