2024icpc江西：H.Convolution(二维前缀和，卷积核)

题目

做法

我们发现，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都是一样规模的矩阵。