灌溉--蓝桥杯（枚举）

灌溉–原题

题目描述

小蓝负责花园的灌溉工作。
花园可以看成一个 nn 行 mm 列的方格图形。中间有一部分位置上安装有出水管。
小蓝可以控制一个按钮同时打开所有的出水管，打开时，有出水管的位置可以被认为已经灌溉好。
每经过一分钟，水就会向四面扩展一个方格，被扩展到的方格可以被认为已经灌溉好。即如果前一分钟某一个方格被灌溉好，则下一分钟它上下左右的四个方格也被灌溉好。
给定花园水管的位置，请问 kk 分钟后，有多少个方格被灌溉好？

输入描述

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

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

接下来 tt 行描述出水管的位置，其中第 ii 行包含两个数 r, cr,c 表示第 rr 行第 cc 列有一个排水管。
接下来一行包含一个整数 kk。
其中，1 \leq n, m \leq 100, 1 \leq t \leq 10, 1 \leq k \leq 1001≤n,m≤100,1≤t≤10,1≤k≤100。

输出描述

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

输入输出样例

输入

3 6
2
2 2
3 4
1

输出

9

运行限制

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

#include <iostream>
#include<bits/stdc++.h>
#define N 5001
#define INF 0x3f3f3f3f
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define rep(i,a,n) for(int i=a;i<=n;i++)
typedef long long ll;
using namespace std;
int a[N][N];
void water(int x,int y)
{
    a[x][y]=1;
    if(!a[x-1][y])a[x-1][y]=1;
    if(!a[x+1][y])a[x+1][y]=1;
    if(!a[x][y-1])a[x][y-1]=1;
    if(!a[x][y+1])a[x][y+1]=1;

}
int main()
{
    int m,n,t,k;
    cin>>n>>m;
    cin>>t;//几个浇水
    while(t--)
    {
        int x,y;
        cin>>x>>y;
        water(x,y);
    }
    cin>>k;
    int g=k-1;
    if(k>1)
    {
        while(g--)
        {
            for(int i=1; i<=n; i++)
                for(int j=1; j<=m; j++)
                {
                    if(a[i][j]==1)
                    {
                        water(i,j);
                    }
                }
        }
    }
    int res=0;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            if(a[i][j]==1)
            {
                res++;
            }
        }
    }
    cout<<res;
    return 0;
}