洛谷 P1514 引水入城

题目描述

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

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

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

输入输出格式

输入格式：

 

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

1
1

输入样例#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 说明】

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

【样例2 说明】

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

【数据范围】

题中共给了两个问题。对于第一个问题我们只需从第一排的每个城市开始拓展，并且将已经拓展到的点标记。然后看最后一排的城市有没有被拓展到。这里用dfs就可以轻松解决。

对于第二个问题，仅仅考虑能够满足条件的情况。我们首先需要知道这样几个事实：我们将不同蓄水厂延伸的路径称为不同的区域，那么最后一排的相邻的不同区域一定是连续的。并且同一个区域必然联通。

那么我们将一个蓄水厂延伸到最后一排的区域看成一个区间，这就转化成了区间覆盖问题。贪心即可解决。

我们用l[i][j],r[i][j]分别表示(i,j)能够延伸到最后一排的左端点和右端点，然后在dfs中回溯的时候更新即可。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int inf=0x3f;
int n,m,s[505][505],maxx,l[505][505],r[505][505],cnt;
bool vis[505][505];
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
void check_ans(int x,int y){
	vis[x][y]=1;
	for(int i=0;i<=3;i++){
		int tx=x+dx[i],ty=y+dy[i];
		if(tx<1||tx>n||ty<1||ty>m||s[tx][ty]>=s[x][y])continue;
		if(!vis[tx][ty])check_ans(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;
	memset(l,inf,sizeof(l));
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)cin>>s[i][j];
	for(int i=1;i<=m;i++)l[n][i]=r[n][i]=i;//最后一排的区间初始化。
	for(int i=1;i<=m;i++)
		if(!vis[1][i])check_ans(1,i);
	for(int i=1;i<=m;i++)
		if(!vis[n][i])cnt++;
	if(cnt){
		puts("0");
		cout<<cnt;
		return 0;
	}
	puts("1");
	int L=1;
	while(L<=m){
		int maxr=0;
		for(int j=1;j<=m;j++)
			if(l[1][j]<=L)maxr=max(maxr,r[1][j]);
		cnt++;
		L=maxr+1;
	}
	cout<<cnt;
}