ccf-csp：202104-2邻域均值

问题描述

 

试题编号：	202104-2
试题名称：	邻域均值
时间限制：	1.0s
内存限制：	512.0MB
问题描述：	试题背景 顿顿在学习了数字图像处理后，想要对手上的一副灰度图像进行降噪处理。不过该图像仅在较暗区域有很多噪点，如果贸然对全图进行降噪，会在抹去噪点的同时也模糊了原有图像。因此顿顿打算先使用邻域均值来判断一个像素是否处于较暗区域，然后仅对处于较暗区域的像素进行降噪处理。 问题描述 待处理的灰度图像长宽皆为 n 个像素，可以表示为一个 n×n 大小的矩阵 A，其中每个元素是一个 [0,L) 范围内的整数，表示对应位置像素的灰度值。 对于矩阵中任意一个元素 Aij（0≤i,j<n），其邻域定义为附近若干元素的集和： Neighbor(i,j,r)={Axy|0≤x,y<n and |x−i|≤r and |y−j|≤r} 这里使用了一个额外的参数 r 来指明 Aij 附近元素的具体范围。根据定义，易知 Neighbor(i,j,r) 最多有 (2r+1)2 个元素。 如果元素 Aij 邻域中所有元素的平均值小于或等于一个给定的阈值 t，我们就认为该元素对应位置的像素处于较暗区域。 下图给出了两个例子，左侧图像的较暗区域在右侧图像中展示为黑色，其余区域展示为白色。 现给定邻域参数 r 和阈值 t，试统计输入灰度图像中有多少像素处于较暗区域。 输入格式 输入共 n+1 行。 输入的第一行包含四个用空格分隔的正整数 n、L、r 和 t，含义如前文所述。 第二到第 n+1 行输入矩阵 A。 第 i+2（0≤i<n）行包含用空格分隔的 n 个整数，依次为 Ai0,Ai1,⋯,Ai(n−1)。 输出格式 输出一个整数，表示输入灰度图像中处于较暗区域的像素总数。 样例输入 4 16 1 6 0 1 2 3  4 5 6 7 8 9 10 11 12 13 14 15 样例输出 7 样例输入 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 样例输出 83 评测用例规模与约定 70% 的测试数据满足 n≤100、r≤10。 全部的测试数据满足 0<n≤600、0<r≤100 且 2≤t<L≤256。

sos，写了快1.5小时

#include<bits/stdc++.h>
using namespace std;
int n,L,r,t,ans=0;
int s[1010][1010];
int a1,a2,b1,b2,a3,a4,b3,b4;

int main()
{
	cin>>n>>L>>r>>t;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>s[i][j];
		}
	}
	
	for(int i=0;i<n;i++)
	{
		int sum=0,j=0,num=0;
		a1=i-r; if(a1<0) a1=0;
		b1=j-r; if(b1<0) b1=0;
		a2=i+r; if(a2>=n) a2=n-1;
		b2=j+r; if(b2>=n) b2=n-1;
		
		for(int x=a1;x<=a2;x++)
		{
			for(int y=b1;y<=b2;y++)
			{
				sum+=s[x][y];
				num++;
			}
		}
		double temp=double(sum)/double(num);
		if(temp<=double(t)) ans++;
		
		for(j=1;j<n;j++)
		{
			a3=i-r; if(a3<0) a3=0;
			b3=j-r; if(b3<0) b3=0;
			a4=i+r; if(a4>=n) a4=n-1;
			b4=j+r; if(b4>=n) b4=n-1;
			if(b1!=b3)
			{
				for(int x=a1;x<=a2;x++)
				{
					sum-=s[x][b1];
					num--;
				}
			}
			if(b2!=b4)
			{
				for(int x=a1;x<=a2;x++)
				{
					sum+=s[x][b4];
					num++;
				}
			}
			double temp=double(sum)/double(num);
			if(temp<=double(t)) ans++;
			
			a1=a3;b1=b3;a2=a4;b2=b4;
		}
	}
	cout << ans;
	return 0;
 }