题目
做法
我们发现,K中的每个元素都乘了对应原矩阵中子矩阵(n-k+1)* (m-l+1)中的每个数。我们就直接前缀和求就好了。
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e3+10;
int n,m,k,l;
int a[N][N],qzh[N][N];
int ans;
signed main(){
scanf("%lld%lld%lld%lld",&n,&m,&k,&l);
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
scanf("%lld",&a[i][j]);
qzh[i][j]=a[i][j]+qzh[i-1][j]+qzh[i][j-1]-qzh[i-1][j-1];
}
}
int nn=n-k+1,mm=m-l+1;//O矩阵的规模
for(int i=1;i+nn-1<=n;i++){
for(int j=1;j+mm-1<=m;j++){
int sum=qzh[i+nn-1][j+mm-1]-qzh[i-1][j+mm-1]-qzh[i+nn-1][j-1]+qzh[i-1][j-1];
if(sum>0) ans+=sum;
else ans+=-sum;
}
}
cout<<ans;
}
WA的原因
样例中K矩阵和O矩阵都是3*3的,我就理所应当的以为K和O都是一样规模的矩阵。