前缀和问题2（c++）

干货

题目练习

（续上一篇，如需了解上一篇题目，请移步主页观看）

和的绝对值不超过 K 的最长子数组

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n,m;
int a[5010][5010];
int dp[5010][5010];
int man,mam;

int main()
{
	cin>>n>>m;
	for(int i = 1;i<=n;i++)
	{
		int x,y,v;
		cin>>x>>y>>v;
		man = max(man,x);
		mam = max(mam,y);
		a[x-1][y-1] = v;
	}
	for(int i = 1;i<=man;i++)
	{
		for(int j = 1;j<=mam;j++)
		{
			dp[i][j] = dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+a[i-1][j-1];
		}
	}
	int ma = -999999999;
	for(int i1 = 0;i1<=man-m;i1++)
	{
		for(int j1 = 0;j1<=man-m;j1++)
		{
			int i2 = i1+m-1;
			int j2 = j1+m-1;
			int sum = dp[i2+1][j2+1]-dp[i2+1][j1]-dp[i1][j2+1]+dp[i1][j1];
			ma = max(ma,sum);
		}
	}
	cout<<ma;
	
	return 0;
}