【题解】洛谷[NOIP2010 提高组] 引水入城

题目链接

P1514

分析

题目的第一问是干旱区能否都能有水流到，我们只需从靠近湖泊一排每一个位置开始DFS，就可以确定答案。考虑到时间复杂度的问题，我们加上记忆化即可。

如果无法满足要求，只需要输出0并遍历标记数组的最后一行一遍累计出有几座干旱区中的城市不可能建有水利设施。但对于能满足要求的情况，如何求出最少需要多少个蓄水厂呢？让我们先来做一个证明：

每个水库的所供给的靠近沙漠的城市，一定是一个区间

这句话不难理解，但是你可能会很快想到这样一个反例：

分叉了不是显而易见的吗？证明个啥啊？！
但是这里要注意了！我们只有在满足条件的情况下才会求这一问题，当出现分叉的情况时，中间高的块是无法流经任何水的，也就是说无法满足条件。
通过上述简单且草率的证明，第二问就被转换成了一个区间覆盖问题。
总结：记忆化搜索+区间覆盖

代码

详细说明请看注释：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

const int N=510;
typedef pair<int,int> PII;
int n,m,a[N][N],st[N][N],l[N][N],r[N][N];
int dx[]={1,-1,0,0},dy[]={0,0,1,-1};//偏移量
void dfs(int x,int y)
{
	st[x][y]=1;//标记流到过
	if(x==n) l[x][y]=r[x][y]=y;
	else
	{
		l[x][y]=m+1;
		r[x][y]=0;
	}//设初值
	for(int i=0;i<4;i++)
	{
		int xx=x+dx[i],yy=y+dy[i];
		if(xx>n||xx<1||yy>m||yy<1) continue;//不能出界
		if(a[xx][yy]>=a[x][y]) continue;//水往低处流
		if(!st[xx][yy])dfs(xx,yy);
		l[x][y]=min(l[x][y],l[xx][yy]);
		r[x][y]=max(r[x][y],r[xx][yy]);//求能覆盖的最右边和最左边
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++) cin>>a[i][j];
	for(int i=1;i<=m;i++) if(!st[1][i])dfs(1,i);//第一排开始流
	int t=0;
	for(int i=1;i<=m;i++) if(!st[n][i]) t++;//求有几座干旱区中的城市不可能建有水利设施
	if(t==0)
	{
		vector<PII> v;
		for(int i=1;i<=m;i++)
			if(r[1][i]>=1&&l[1][i]<=m) v.push_back({l[1][i],r[1][i]+1});//给各个区间的左右端点赋值
		sort(v.begin(),v.end());//根据左端点排序
		int idx=1,mx=0;
		for(int i=0;i<v.size();i++)
		{
			if(v[i].first<=idx)
			{
				mx=max(mx,v[i].second);
				continue;
			}//当左端点都能覆盖到某一位置，就尽可能让右端点更大
			idx=mx;
			t++;
			mx=max(mx,v[i].second);
			if(mx>m) break;
		}
		if(idx<=m) t++;//如果没覆盖完就还需要再建一个，我因为这点WA了好久
		cout<<1<<endl<<t;
	}
	else cout<<0<<endl<<t;
	return 0;
}

最后的话

虽然这题核心在于上述结论的证明，记忆化搜索和区间覆盖的代码实现不难，但是小细节很容易出错，调起来有一些些恶心，所以不建议练习时进行口胡。