最大子矩阵

给你一个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

 

题解: dp[i][j] 代表以 i 行 j 列的元素作为右下角的矩阵的和

dp完后，容斥找到答案

 1 #include <iostream>
 2 #include <stdio.h>
 3 #include <string.h>
 4 using namespace std;
 5 #define MAXN 1005
 6 int n,m,x,y;
 7 int data[MAXN][MAXN];
 8 int dp[MAXN][MAXN];
 9 
10 int main()
11 {
12     int T;
13     cin>>T;
14     while (T--)
15     {
16         scanf("%d%d",&n,&m);
17         scanf("%d%d",&x,&y);
18         for (int i=0;i<n;i++)
19            for (int j=0;j<m;j++)
20                 scanf("%d",&data[i][j]);
21         for (int i=0;i<n;i++)
22         {
23             for (int j=0;j<m;j++)
24             {
25                 dp[i][j]=data[i][j];
26                 if (j-1>=0)
27                     dp[i][j]+=dp[i][j-1];
28                 if (i-1>=0)
29                     dp[i][j]+=dp[i-1][j];
30                 if (i-1>=0&&j-1>=0)
31                     dp[i][j]-=dp[i-1][j-1];
32             }
33         }
34 
35         int ans = 0;
36         for (int i=x-1;i<n;i++)
37         {
38             for (int j=y-1;j<m;j++)
39             {
40                 int res = dp[i][j];
41                 if (i-x>=0)
42                     res -= dp[i-x][j];
43                 if (j-y>=0)
44                     res -= dp[i][j-y];
45                 if (i-x>=0&&j-y>=0)
46                     res += dp[i-x][j-y];
47                 if (res>ans)
48                     ans = res;
49             }
50         }
51         printf("%d\n",ans);
52     }
53     return 0;
54 }

View Code

 

转载于:https://www.cnblogs.com/haoabcd2010/p/6748433.html