2013年4月_武科大程序设计大赛(复赛)_解题报告(problem 1004)

最大子矩阵

Time Limit: 30000/10000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 28    Accepted Submission(s): 2

Problem Description

  
  

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

 

Author

  
  

   
   lwg
  
  
  
  


  
  
  
  

   
   感谢波波学长的代码，直接枚举。
  
  
  
  

   
   
#include<iostream>
#include<cstdio>

using namespace std;
const int N=1005;

int matrix[N][N];
int s[N];//临时数组,保留上次计算结果

int main()
{
    int t,m,n,x,y,max;
    int i,j,k;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d%d",&m,&n,&x,&y);
        for(i=1;i<=m;i++)
            for(j=1;j<=n;j++)
                scanf("%d",&matrix[i][j]);
        //计算右上坐标为（1，i）的所有子矩阵和(初始化)
        s[y]=0;
        for(i=1;i<=x;i++)
            for(j=1;j<=y;j++)
                s[y]+=matrix[i][j];

        max=s[y];
        for(i=y+1;i<=n;i++)
        {
            //temp[0]记录第i-y列，temp[1]记录第i列
            int temp[2]={0,0};
            for(j=1;j<=x;j++)
            {
                temp[0]+=matrix[j][i-y];
                temp[1]+=matrix[j][i];
            }
            s[i]=s[i-1]-temp[0]+temp[1];
            if(s[i]>max)
                max=s[i];

        }
        //计算其他子矩阵，要充分利用前次计算结果
        for(i=x+1;i<=m;i++)
        {
            for(j=y;j<=n;j++)
            {
                //temp[0]记录第i-x行，temp[1]记录第i行
                int temp[2]={0,0};
                for(k=j-y+1;k<=j;k++)
                {
                    temp[0]+=matrix[i-x][k];
                    temp[1]+=matrix[i][k];
                }
                //更新s[j]
                s[j]=s[j]-temp[0]+temp[1];
                if(s[j]>max)
                    max=s[j];
            }
        }
        printf("%d\n",max);
    }
    return 0;
}
   
   

    
    

   
   
感谢坤坤的代码，看完后很巧妙的感觉。
  
  
  
  

   
   读入数据的时候将以i,j为右下角0,0为左上角的矩形的和存入a中，
  
  
  
  

   
   为了能边读入边存,可以人为的在矩阵周围加上一圈0
  
  
  
  

   
   0   0   0   0   0   0 
   
   
0   3   361 649 676 588
   
   
0   992 762 156 993 169
   
   
0   662 34  638 89  543
   
   
0   525 165 254 809 280
  
  
  
  

   
   例如a[4][4]中存入的就是 3+361+649+992+762+156+662+34+638;
  
  
  
  

   
   而要求x*y矩阵的和就只要用 a[i][j]-a[i][j-y]-a[i-x][j] + a[i-x][j-y];
  
  
  
  

   
   (左上角的数被减去了两次)。
  
  
  
  

   
   
   
   
# include<stdio.h>
# include<string.h>
int a[1005][1005],b[1005][1005];
int main()
{
    int t,x,y,n,m,i,j,s,max;
    scanf("%d",&t);
    while (t--)
    {
        memset(a,0,sizeof(a));
        scanf("%d%d%d%d",&m,&n,&x,&y);
        for (i=1;i<=m;i++)
            for (j=1;j<=n;j++)
            {
                scanf("%d",&a[i][j]);
                a[i][j]=a[i][j]-a[i-1][j-1]+a[i][j-1]+a[i-1][j];
            }
        max=0;
        memset(b,0,sizeof(b));
        for (i=x;i<=m;i++)
            for (j=y;j<=n;j++)
            {
                s=a[i][j]+a[i-x][j-y]-a[i][j-y]-a[i-x][j];
                if (max<s) max=s;
            }
        printf("%d\n",max);
    }
    return 0;
}