题目:
样例输入1
4 16 1 6
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15
样例输出1
7
样例输入2
11 8 2 2
0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0
0 7 0 0 0 7 0 0 7 7 0
7 0 7 0 7 0 7 0 7 0 7
7 0 0 0 7 0 0 0 7 0 7
7 0 0 0 0 7 0 0 7 7 0
7 0 0 0 0 0 7 0 7 0 0
7 0 7 0 7 0 7 0 7 0 0
0 7 0 0 0 7 0 0 7 0 0
0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0
样例输出2
83
问题规模
问题分析:
这道题的解题思路:
1.求出来元素周围半径为r的内容的和sum
2.对内容和求平均值px
3.将px与t进行比较
分析复杂度:
对于第一部分,我们第一反应是打暴力,对于n<=100,r<=10的情况,时间复杂度为:1001001010=1e6,可行;对于0<n<=600,0<r<=100的情况,时间复杂度为600600100100=3.6e9,显然是会TLE的,所以需要优化
对于第二三部分,时间复杂度都为O(1),没有优化空间
所以我们优化的重点就在第1步
对于求和的优化,首先想到的就是前缀和,此处用到了二维前缀和
对于二维前缀和的讲解请看这里
对样例进行分析
可以看到,当r=1时,对于 5 [坐标(2,2)]来说,关键的点为 10 [坐标(3,3)]和 0 [坐标(1,1)]
不难分析出,对于点(i,j)来说,其邻域为对角从点(i-r,j-r)到点(i+r,j+r)围起来的正方形区域
同时,要对边界情况进行处理,当 i-r<1 或 j-r<1 时,要取为1,当 i+r>n 或 j+r>n 时,要取为n
以点(1,1)为例,其i-r=0<1,故取1
代码:
#include <bits/stdc++.h>
using namespace std;
int n,l,r;
double t;
int a[605][605];
int b[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];
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j];//预处理
}
}
int ans=0;
int sum=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
int x1,x2,y1,y2;
x1=max(i-r,1);
y1=max(j-r,1);
x2=min(i+r,n);
y2=min(j+r,n);//对越界情况进行处理
sum=b[x2][y2]-b[x1-1][y2]-b[x2][y1-1]+b[x1-1][y1-1];//求和
// printf("%d ",sum);
int num=(x2-x1+1)*(y2-y1+1);//求个数
double px=(double)sum/(double)num;//求均值
// printf("%.2lf ",px);
if(px<=t)
{
ans++;//记录小于等于t的个数
}
}
// printf("\n");
}
printf("%d\n",ans);
return 0;
}