SDNU_ACM_ICPC_2020_Winter_Practice_2nd A 前缀和二维

题目

给你一个m×n的整数矩阵，在上面找一个x×y的子矩阵，使子矩阵中所有元素的和最大。
Input
输入数据的第一行为一个正整数T，表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y（0<m,n<1000 AND 0<x<=m AND 0<y<=n），表示给定的矩形有m行n列。接下来这个矩阵，有m行，每行有n个不大于1000的正整数。
Output
对于每组数据，输出一个整数，表示子矩阵的最大和。
Sample Input
1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280
Sample Output
2474

思路

AC代码

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<string.h>
#include<vector>
#include<cmath>
#define mod 998244353
using namespace std;
typedef long long ll;
int s[1010][1010];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(s,0,sizeof(s));
        int x,y,m,n;
        int p;
        scanf("%d %d %d %d",&m,&n,&x,&y);
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&p);
                s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+p;///第i行从1~j个数的和
            }
        }
        int ans=0;
        for(int i=1;i<=m-x+1;i++)
        {
            for(int j=1;j<=n-y+1;j++)
            {
                ans=max(ans,s[i+x-1][j+y-1]-s[i-1][j+y-1]-s[i+x-1][j-1]+s[i-1][j-1]);
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}