CCF-CSP认证 202104-2邻域均值
题目描述
思路
二维数组前缀和,画图解决,需要理清楚多增加的元素,本题索引应从1开始,相当于多增加一行和一列0元素,注意边界。
代码
C++版:
#include <bits/stdc++.h>
using namespace std;
const int N = 610;
int main(){
int n,L,r,t;
int res=0; // 统计结果
cin>>n>>L>>r>>t; // 图像长宽,元素值范围上限,附近元素泛围,所有附近元素的平均值小于或等于的阈值
// vector<vector<int> > A(n+1,vector<int>(n+1,0)); // 二维数组前缀和
int A[N][N];
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
int x;
cin>>x;
A[i][j]=A[i-1][j]+A[i][j-1]+x-A[i-1][j-1];
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
// 右下角,小心边界
int x1=min(n,i+r);
int y1=min(n,j+r);
// 左上角,小心边界
int x2=max(1,i-r);
int y2=max(1,j-r);
int sum=A[x1][y1]-A[x1][y2-1]-A[x2-1][y1]+A[x2-1][y2-1];
int k=(x1-x2+1)*(y1-y2+1); // 元素数量
if(sum<=k*t) res++;
}
}
cout<<res;
return 0;
}