题意:给定一个矩阵,求一个宽为a,长为b的子矩阵的最大和
分析:直接用最大和子矩阵的公式,只是这里多了一个限制条件,所以在求和的时候稍有点不同,就是不逐行和逐列求了而是每次以a,b为跨度来求
dp[k]=mt[i][k]+....+mt[i+a][k],然后每次求sum=dp[k]+.....+dp[k+b],mx=max(mx,sum)
代码:
#include<iostream> #include<cstring> using namespace std; int m,n,a,b,t; int mt[1005][1005]; int mx,sum,dp[1005]; int max(int i,int j) { return i>j?i:j; } void DP() { mx=0; for(int i=0;i<m;i++){ memset(dp,0,sizeof(dp)); for(int j=0;j<a&&i+j<m;j++){ for(int k=0;k<n;k++){ dp[k]+=mt[i+j][k]; } } sum=0; for(int k=0;k<b;k++) sum+=dp[k]; mx=max(mx,sum); for(int k=b;k<n;k++){ sum-=dp[k-b]; sum+=dp[k]; mx=max(mx,sum); } } } int main() { cin>>t; while(t--){ cin>>m>>n>>a>>b; for(int i=0;i<m;i++) for(int j=0;j<n;j++) cin>>mt[i][j]; DP(); cout<<mx<<endl; } }
时间: 2024-10-12 16:21:17