P1514 [NOIP2010 提高组] 引水入城 （记忆化搜索）

第一道蓝题纪念
原题

---

[NOIP2010 提高组] 引水入城

题目背景

NOIP2010 提高组 T4

题目描述

在一个遥远的国度，一侧是风景秀美的湖泊，另一侧则是漫无边际的沙漠。该国的行政区划十分特殊，刚好构成一个 $N$ 行 $M$ 列的矩形，如上图所示，其中每个格子都代表一座城市，每座城市都有一个海拔高度。

为了使居民们都尽可能饮用到清澈的湖水，现在要在某些城市建造水利设施。水利设施有两种，分别为蓄水厂和输水站。蓄水厂的功能是利用水泵将湖泊中的水抽取到所在城市的蓄水池中。

因此，只有与湖泊毗邻的第 $1$ 行的城市可以建造蓄水厂。而输水站的功能则是通过输水管线利用高度落差，将湖水从高处向低处输送。故一座城市能建造输水站的前提，是存在比它海拔更高且拥有公共边的相邻城市，已经建有水利设施。由于第 $N$ 行的城市靠近沙漠，是该国的干旱区，所以要求其中的每座城市都建有水利设施。那么，这个要求能否满足呢？如果能，请计算最少建造几个蓄水厂；如果不能，求干旱区中不可能建有水利设施的城市数目。

输入格式

每行两个数，之间用一个空格隔开。输入的第一行是两个正整数 $N,M$，表示矩形的规模。接下来 $N$ 行，每行 $M$ 个正整数，依次代表每座城市的海拔高度。

输出格式

两行。如果能满足要求，输出的第一行是整数 $1$，第二行是一个整数，代表最少建造几个蓄水厂；如果不能满足要求，输出的第一行是整数 $0$，第二行是一个整数，代表有几座干旱区中的城市不可能建有水利设施。

样例 #1

样例输入 #1

2 5
9 1 5 4 3
8 7 6 1 2

样例输出 #1

1
1

样例 #2

样例输入 #2

3 6
8 4 5 6 4 4
7 3 4 3 3 3
3 2 2 1 1 2

样例输出 #2

1
3

提示

样例 1 说明

只需要在海拔为 $9$ 的那座城市中建造蓄水厂，即可满足要求。

样例 2 说明

上图中，在 $3$ 个粗线框出的城市中建造蓄水厂，可以满足要求。以这 $3$ 个蓄水厂为源头在干旱区中建造的输水站分别用 $3$ 种颜色标出。当然，建造方法可能不唯一。

---

首先明确只有最上面一行可以产生水流，并且我们需要让最后一行全部能够被水流经过。
这时候应该要想到一个做法就是：

• 先去搜索上面所有点，然后得到每个点单独流水下来能够覆盖到的最后一行的区间，然后进行贪心的区间合并就好了。

但是如果我们暴力的去搜的话复杂度会很大，所以这时候我们需要使用记忆化搜索对搜索进行一个剪枝。

我们的目的是得到左右边界的区间，所以我们直接使用两个dp数组，一个数组l[i][j]代表从点$(i,j)$开始往四周流水流下来能够覆盖到的最后一行的左边界，另一个r[i][j]同理，我们只需要这样进行维护就可以使用判重数组直接对搜过的点进行判重了，只有这样能够不取漏下任何一个点。

至于更新数组的方法，我们对于左边界去维护最小值，右边界维护最大值，然后更新的方法是：

• 我们先使用dfs去深入的搜索到最后一层，这样一层层的从下面往上面递归，就可以动态的维护住上面的点位能够达到的区间。

CODE:

int g[N][N];
int l[N][N];    //l[i][j]代表从(i,j)开始往外扩展流水到最后能够达到的左边界
int r[N][N];
bool vis[N][N];
int n, m;

void dfs(int x, int y)
{
    vis[x][y] = 1;

    for (int i = 0; i < 4; i++)
    {
        int a = x + dx[i], b = y + dy[i];
        if (a < 1 || a > n || b < 1 || b > m)
            continue;
        if (g[a][b] >= g[x][y])
            continue;
        if(!vis[a][b])dfs(a, b);        //先去搜下一个点
        l[x][y] = min(l[a][b],l[x][y]); //搜完后面的层数之后层层往上进行更新
        r[x][y] = max(r[x][y],r[a][b]);
    }
}

void solve(int times)
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            cin >> g[i][j];
        }
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            l[i][j] = j;
            r[i][j] = j;
        }
    }

    for (int i = 1; i <= m; i++) dfs(1, i);

    int cnt = 0;
    for (int i = 1; i <= m; i++)
    {
        if (vis[n][i])
            cnt++;
    }
    int res = 0;
    if (cnt < m)
    {
        cout << 0 << endl << m - cnt << endl;
    }
    else
    {
        int L = 1;
        while (L <= m)
        {
            int maxr = 0;
            for (int i = 1; i <= m; i++)
            {
                if (l[1][i] <= L)       //l[1][i]代表从(1,i)这个点流水流到最后能够得到的左边界
                    maxr = max(maxr, r[1][i]);  //r[1][i]代表从(1,i)这个点流水流到最后能够得到的右边界
            }
            res++;
            L = maxr + 1;
        }
        cout << 1 << endl << res << endl;
    }
}