PTA 引水入城

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

为了使居民们都尽可能饮用到清澈的湖水，现在要在某些城市建造水利设施。水利设施有两种，分别为蓄水厂和输水站。蓄水厂的功能是利用水泵将湖泊中的水抽取到所在城市的蓄水池中。因此，只有与湖泊毗邻的第 1 行的城市可以建造蓄水厂。而输水站的功能则是通过输水管线利用高度落差，将湖水从高处向低处输送。故一座城市能建造输水站的前提，是存在比它海拔更高且拥有公共边的相邻城市，已经建有水利设施。

由于第 N 行的城市靠近沙漠，是该国的干旱区，所以要求其中的每座城市都建有水利设施。那么，这个要求能否满足呢？如果能，请计算少建造几个蓄水厂；如果不能，求干旱区中不可能建有水利设施的城市数目。

输入格式:

输入的每行中两个数之间用一个空格隔开。

输入的第一行是两个正整数 N 和 M，表示矩形的规模；

接下来 N 行，每行 M 个正整数，依次代表每座城市的海拔高度。

输出格式:

输出有两行。

如果能满足要求，输出的第一行是整数 1 ，第二行是一个整数，代表最少建造几个蓄水厂；

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

输入样例1:

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

输出样例1:

1
1

样例说明1:

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

输入样例2:

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

输出样例2:

1
3

样例说明2:

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

数据范围

 思路：

定义一个l 和 r数组，记录第一层最多能到下方的最左端和最右端。

如果中间会有跑不到的点的话，会在第一次遍历最后一层有无到达中去掉。

    for(int i = 1 ; i <= m ; i ++)
		if(st[n][i] == 0)sum ++; 

	if(sum){
		cout << 0 << endl << sum << endl;
		return 0;
	}

否则一定能从那一次到达最小左端点和最大右端点。

初始每个l[i][j] 为m+1 ，r[i][j] 初始为0。

最后一层的每个点都能到自身: l[n][i] = i , r[n][i] = i;

    for(int i = 1 ; i <= n ; i ++)
		for(int j = 1 ; j <= m ; j ++)
			cin >> a[i][j],l[i][j] = m+1;

	for(int i = 1 ; i <= m ; i ++){
		l[n][i] = r[n][i] = i;
	}

在记忆化深搜中，我们会先向下进行搜索，向上递归每个点能到的最左最右端点。

ps：初始最后一层都能到自身，重要的的事情说两遍。

void dfs(int x,int y){
	st[x][y] = 1;
	for(auto i : dir){
		int tx = x + i[0];
		int ty = y + i[1];
		if(tx < 1 || ty < 1 || tx > n || ty > m || a[tx][ty] >= a[x][y] )continue;
		if(!st[tx][ty])dfs(tx,ty);
		l[x][y] = min(l[x][y],l[tx][ty]);r[x][y] = max(r[x][y],r[tx][ty]);
	}
}

如果判断最后一层都能跑到，那么就遍历左端点能到当前点，此区间能到的最大右边端点。

    sum = 0;
	int now = 1;
	while(now !=  m + 1){
		int next = now - 1;
		for(int i = 1 ; i <= m ; i ++){
			if(l[1][i] <= now){
				next = max(r[1][i],next);
			}
		}
		sum ++;
		now = next + 1;
	}

总代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n , m , T ;
const int N = 505;
int l[N][N] , r[N][N] , a[N][N] , st[N][N];
int dir[4][2] = {0,1,1,0,0,-1,-1,0};

void dfs(int x,int y){
	st[x][y] = 1;
	for(auto i : dir){
		int tx = x + i[0];
		int ty = y + i[1];
		if(tx < 1 || ty < 1 || tx > n || ty > m || a[tx][ty] >= a[x][y] )continue;
		if(!st[tx][ty])dfs(tx,ty);
		l[x][y] = min(l[x][y],l[tx][ty]);r[x][y] = max(r[x][y],r[tx][ty]);
	}
}

int main(){
	cin >> n >> m;
	for(int i = 1 ; i <= n ; i ++)
		for(int j = 1 ; j <= m ; j ++)
			cin >> a[i][j],l[i][j] = m+1;

	for(int i = 1 ; i <= m ; i ++){
		l[n][i] = r[n][i] = i;
	}

	for(int i = 1 ; i <= m ; i ++){
		if(!st[1][i])dfs(1,i);
	}

	int sum = 0;

	for(int i = 1 ; i <= m ; i ++)
		if(st[n][i] == 0)sum ++; 

	if(sum){
		cout << 0 << endl << sum << endl;
		return 0;
	}
	sum = 0;
	int now = 1;
	while(now !=  m + 1){
		int next = now - 1;
		for(int i = 1 ; i <= m ; i ++){
			if(l[1][i] <= now){
				next = max(r[1][i],next);
			}
		}
		sum ++;
		now = next + 1;
	}
	
	cout << 1 << endl << sum << endl;return 0;
	
	sum = 0;
	
}