CCF-CSP 202104-2 邻域均值

 

 题目很冗长，但是实际上内含的算法并不难，主要注意理解题意，

该公式表明领域的定义在于以中心点展开边长为r的正方形，然后我们需要求出该正方形内所有元素之和并且求出其平均值，然后对平均值<=t的进行计数即可，首先我们可以确定的是对中心点的遍历，即首先必然需要一个嵌套循环，然后后续求平均值，我们可以再嵌套循环进行求解，但是我们有更好的算法：二维前缀和，这样总体时间复杂度就为O(n^2)

便解决了

 

#include<iostream>
#include<cmath>
using namespace std;

int n,L,r,t,ans=0;
int A[605][605];
int pre[605][605];//前缀和数组 

int main(){
	cin>>n>>L>>r>>t;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>A[i][j];
			pre[i][j]=A[i][j]+pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]; 
		}
	}
//	for(int i=1;i<=n;i++){
//		for(int j=1;j<=n;j++){//调试
//			cout<<pre[i][j]<<" ";
//		}
//		cout<<endl;
//	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			double sum=0.0,count=0.0,avg;
			int st1=i-r<1?1:i-r,ed1=i+r>n?n:i+r;//小心越界 
			int st2=j-r<1?1:j-r,ed2=j+r>n?n:j+r;
			//右下角（ed1,ed2） 左上角（st1,st2）
			sum=pre[ed1][ed2]-pre[ed1][st2-1]-pre[st1-1][ed2]+pre[st1-1][st2-1];
			count=(ed1-st1+1)*(ed2-st2+1);//注意这里的sum别写错了
			avg=sum/count;
			if(avg<=t){
				ans++;
			} 
		}
	}
	cout<<ans<<endl;
	return 0;
}