问题描述
**
解题思路
**
这道题70%的数据比较小,可以采用暴力的方式直接算出结果。但剩余30%的测试数据会超时,只能得到70分。因此我们可以采用前缀和的方式存储矩阵中的内容,来解决这道题。
二维数组A[i][j]接收输入的矩阵,s[601][601]记录前缀和,其中s[i][j]表示在数组A中从第1行到第i行,每行第1到第j个总共i*j个元素的和,即从这个位置开始,向左上方构造出的矩阵的所有元素的总和(如s[x][y]的值即为矩阵A中(0,0)点与(x,y)构成矩阵中所有元素的和)。如果要求矩阵A中(i,j)和(x,y)这两点构成的矩阵里的所有元素之和,带入表达式s[x][y]-s[i-1][y]-s[x][j-1]+s[i-1][j-1]即可。
如图,右下角矩阵的面积可通过整个矩阵的面积减去蓝色边框矩阵,再减去绿色边框矩阵的面积,再加上蓝绿边框矩阵相交部分的面积求得。利用数组s可以轻易求出A中任意矩阵的面积。
构造二维数组s的方法与以上过程类似。
AC代码(C++)
#include<bits/stdc++.h>
using namespace std;
int n,l,r,t,A[601][601],s[601][601];
void solve(){
int ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){//将邻域转化为矩阵的形式 (i_min,j_min) (i_max,j_max)为这一位置邻域矩阵的边界
double cal=0;
int i_max=min(n,i+r);
int i_min=max(1,i-r)-1;
int j_max=min(n,j+r);
int j_min=max(1,j-r)-1;
double sum=s[i_max][j_max]-s[i_max][j_min]-s[i_min][j_max]+s[i_min][j_min];
int num=(i_max-i_min)*(j_max-j_min);//矩阵中元素数量
cal=(double)sum/(double)num;//平均值一定要保留小数
if(cal<=t) ans++;
}
}
cout<<ans<<endl;
}
int main()
{
ios::sync_with_stdio(false);
cin>>n>>l>>r>>t;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cin>>A[i][j];
}
}
for(int i=1;i<=n;i++){//构造前缀和数组
for(int j=1;j<=n;j++){
s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+A[i][j];
}
}
solve();
system("pause");
return 0;
}