蓝桥杯官网练习题（灌溉）

题目描述

小蓝负责花园的灌溉工作。

花园可以看成一个 n 行 m 列的方格图形。中间有一部分位置上安装有出水管。

小蓝可以控制一个按钮同时打开所有的出水管，打开时，有出水管的位置可以被认为已经灌溉好。

每经过一分钟，水就会向四面扩展一个方格，被扩展到的方格可以被认为已经灌溉好。即如果前一分钟某一个方格被灌溉好，则下一分钟它上下左右的四个方格也被灌溉好。

给定花园水管的位置，请问 k 分钟后，有多少个方格被灌溉好？

输入描述

输入的第一行包含两个整数 n,m。

第二行包含一个整数 t，表示出水管的数量。

接下来 t 行描述出水管的位置，其中第 i 行包含两个数 r,c 表示第 r 行第 c 列有一个排水管。

接下来一行包含一个整数 k。

其中，1≤n,m≤100,1≤t≤10,1≤k≤100。

输出描述

输出一个整数，表示答案。

输入输出样例

示例 1

输入

3 6
2
2 2
3 4
1

输出

9

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        int ans=0;
        Scanner scan = new Scanner(System.in);
        int n=scan.nextInt();
        int m=scan.nextInt();
        int[][] a=new int[n+1][m+1];
        int h=scan.nextInt();
        for(int i=1;i<=h;i++){
            int r=scan.nextInt();
            int c=scan.nextInt();
            a[r][c]=1;
        }
        int count=1;
        int time=scan.nextInt();
        scan.close();
        for(int i=1;i<=time;i++){
            for(int j=1;j<=n;j++){
                for(int k=1;k<=m;k++){
                    if(a[j][k]==count){
                        if(j-1>=1&&a[j-1][k]==0){a[j-1][k]=count+1;}
                        if(j+1<=n&&a[j+1][k]==0){a[j+1][k]=count+1;}
                        if(k-1>=1&&a[j][k-1]==0){a[j][k-1]=count+1;}
                        if(k+1<=m&&a[j][k+1]==0){a[j][k+1]=count+1;}

                    }
                }
            }
            count++;
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                if(a[i][j]!=0){
                    ans++;
                }
            }
        }
        System.out.println(ans);
    }
}