给你一个m×n的整数矩阵,在上面找一个x×y的子矩阵,使子矩阵中所有元素的和最大。
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
int arr[1010][1010], m, n, x, y;
int lowbit(int x)
{
return x & -x;
}
///二维树状数组
void add(int x, int y, int tt) ///单点修改
{
for (int i = x; i < m; i += lowbit(i))
for (int j = y; j < n; j += lowbit(j))
{
arr[i][j] += tt;
}
}
int sum(int x, int y)
{
int res = 0;
for (int i = x; i; i -= lowbit(i))
for (int j = y; j; j -= lowbit(j))
{
res += arr[i][j];
}
return res;
}
int query(int L, int B, int R, int T) ///矩阵求和
{
return sum(R, T) + sum(L - 1, B - 1) - sum(R, B - 1) - sum(L - 1, T);
}
int main()
{
int t;
scanf("%d", &t);
while (t--){
scanf("%d%d%d%d", &m, &n, &x, &y);
for (int i = 1; i <= m; i++){
for (int j = 1; j <= n; j++){
arr[i][j] = 0;
}
}
int k, ans = 0, gg;
for (int i = 1; i <= m; i++){
for (int j = 1; j <= n; j++){
scanf("%d", &k);
add(i, j, k);
}
}
for (int i = 1; i + x - 1 <= m; i++){
for (int j = 1; j + y - 1 <= n; j++){
gg = query(i, j, i + x - 1, j + y - 1);
ans = max(ans, gg);
}
}
printf("%d\n", ans);
}
return 0;
}
原文:https://www.cnblogs.com/lyj1/p/11915472.html