CCF：202104-2 邻域均值

题目

样例输入与输出1

输入1
4 16 1 6
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15
输出1
17

样例输入与输出2

输入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

题目思路
这道题用人话翻译过来就是：给定一个矩阵，再给定一个坐标(x,y)和一个长度r，求在(x-r,y-r)和(x+r,y+r)这个范围内的元素的和的平均值，是否小于等于给定的t。

读懂题，就能看出来这是一道二维前缀和的问题。

Java代码

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main{
	
	public static void main(String[] args) throws Exception{
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in)); // BufferedReader读入的速度会比Scanner快
		String[] nlrt = bf.readLine().split(" ");
		Integer n = Integer.parseInt(nlrt[0]);
		Integer L = Integer.parseInt(nlrt[1]);
		Integer r = Integer.parseInt(nlrt[2]);
		Integer t = Integer.parseInt(nlrt[3]);
		
		int[][] a = new int[n+2][n+2];
		int[][] sum = new int[n+2][n+2];
		for(int i=1;i<=n;i++) {
			String[] tmp = bf.readLine().split(" ");
			for(int j=1;j<=n;j++) {
				a[i][j]=Integer.parseInt(tmp[j-1]);
				sum[i][j] = sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1] + a[i][j]; // sum数组用来存前缀和
			}
		}
		
		int cnt = 0;
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				int x1 = Math.max(1,i-r);
				int y1 = Math.max(1, j-r);
				int x2 = Math.min(n, i+r);
				int y2 = Math.min(n, j+r);
				if(prefixSum(x1,y1,x2,y2,sum) <= t*getCount(x1,y1,x2,y2)) { // 在r范围内的元素的和/数量 <= 给定的t
					cnt++;
				}
			}
		}
		System.out.println(cnt);
	}

	public static int prefixSum(int x1,int y1,int x2, int y2,int[][] sum) {
		return sum[x2][y2] - sum[x2][y1-1] - sum[x1-1][y2] + sum[x1-1][y1-1];
	}
	
	public static int getCount(int x1,int y1,int x2,int y2) {
		return (x2-x1+1)*(y2-y1+1);
	}
}