CCF-CSP 202104-2 邻域均值 前缀和算法两种方法 满分题解

CCF-CSP 202104-2 邻域均值 前缀和算法两种方法 满分题解

题目链接：202104-2 邻域均值

思路：

• 根据题意，需要求一个点附近r范围内矩阵的和，即范围为(2r+1)²大小的矩阵，考虑使用二维前缀和算法
• 根据前缀和矩阵和行前缀和，分别有两种思路；
• 由于矩阵有边界，因此需要对边界范围进行处理：当要下溢出时，将边界设置为1；当要上溢出时，将边界设置为n
• 矩阵的大小为(x2-x1+1)*(y2-y1+1)
• 小细节：如果直接算均值，均值可能为小数，此处可以改写一些判断条件

方法1前缀和矩阵具体代码：

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e3+10;
int n,l,r,t;
int a[N][N];//输入矩阵
int b[N][N];//前缀和矩阵
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 sum=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            //判断矩阵的边界
            int x1=max(i-r,1);
            int y1=max(j-r,1);
            int x2=min(i+r,n);
            int y2=min(j+r,n);
            
            int temp=(x2-x1+1)*(y2-y1+1);//判断矩阵中点的个数
            if(b[x2][y2]-b[x1-1][y2]-b[x2][y1-1]+b[x1-1][y1-1]<=temp*t)//利用前缀和矩阵求矩阵的和
            {
                sum++;
            }
        }
    }
    cout<<sum<<endl;
    return 0;
}

---

方法2行前缀和具体代码：

//行前缀和
#include <iostream>
#include <math.h>
using namespace std;
const int N = 1e3+10;
int n,l,r,t;
int a[N][N],b[N][N];
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];
            b[i][j]=b[i][j-1]+a[i][j];//行前缀和预处理
        }
    }
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            //边界判断
            int y1 = max(1,j-r);
            int y2 = min(n,j+r);
            int x1 = max(1,i-r);
            int x2 = min(n,i+r);
            int t2 = 0;
            for (int k =x1;k<=x2;k++)
               t2+=b[k][y2]-b[k][y1 - 1];//分别求行前缀和的和
            
            if (t2 <= (x2-x1+ 1) * (y2 - y1 + 1) * t) sum++;
            
        }
    }
    cout<<sum<<endl;
    return 0;
}