CCF 202104-2 试题名称： 邻域均值

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int n;//长宽皆为n个像素，即n*n
    int L;//每个像素的范围[0，L）
    int r;//邻域半径
    int t;//给定阈值

    cin>>n>>L>>r>>t;

    vector<vector<int>> arr;//二维vector数组存放像素点

    for(int i=0;i<n;i++)
    {
        vector<int> v;
        for(int j=0;j<n;j++)
        {
            int tmp;
            cin>>tmp;
            v.push_back(tmp);
        }
        arr.push_back(v);
    }



    int count_an=0;//处于较暗区域的点


    /*思路：暴力遍历的话数据量太大会导致超时
            因为每次移动都有很多重复的数据参与计算
            所以这里使用一个滑动窗口
            每次只更新不同的数据，保留相同的数据
            */
    for(int i=0;i<n;i++)
    {
        int sum=0;
        for(int j=0;j<n;j++)
        {
            int ave=0;
            int xr,xl;
            int yd,yu;
            yu=max(0,i-r);//窗口上边界
            yd=min(n-1,i+r);//窗口下边界
            xl=min(n-1,j+r);//窗口左边界
            xr=max(0,j-r);//窗口右边界
            //cout<<yu<<" "<<yd<<" "<<xr<<" "<<xl<<" ";

            int count=(xl-xr+1)*(yd-yu+1);//窗口包含数据的总个数
            if(j==0)//只在每一行开始的时候进行遍历加法运算，之后都是用滑动窗口
            {
                for(int y=yu;y<=yd;y++)
                {
                    for(int x=xr;x<=xl;x++)
                    {
                        //cout<<arr[y][x]<<" ";
                        sum+=arr[y][x];
                    }
                }
            }
            else
            {
                if(xr==0 && j+r<=n-1)
                {
                    for(int y=yu;y<=yd;y++)
                    {
                        sum+=arr[y][xl];
                    }
                }
                else if(xr!=0 && j+r<=n-1)
                {
                    for(int y=yu;y<=yd;y++)
                    {
                        sum-=arr[y][xr-1];
                        sum+=arr[y][xl];
                    }
                }
                else if(xr!=0 && xl==n-1)
                {
                    for(int y=yu;y<=yd;y++)
                    {
                        sum-=arr[y][xr-1];
                    }
                }

            }

            if(sum<=count*t)
                count_an++;
        }

    }
    cout<<count_an;
    return 0;
}