Luogu 1514 引水入城 (搜索,动态规划)
Description
在一个遥远的国度,一侧是风景秀美的湖泊,另一侧则是漫无边际的沙漠。该国的行政区划十分特殊,刚好构成一个N行M列的矩形,如上图所示,其中每个格子都代表一座城市,每座城市都有一个海拔高度。
为了使居民们都尽可能饮用到清澈的湖水,现在要在某些城市建造水利设施。水利设施有两种,分别为蓄水厂和输水站。蓄水厂的功能是利用水泵将湖泊中的水抽取到所在城市的蓄水池中。
因此,只有与湖泊毗邻的第1行的城市可以建造蓄水厂。而输水站的功能则是通过输水管线利用高度落差,将湖水从高处向低处输送。故一座城市能建造输水站的前提,是存在比它海拔更高且拥有公共边的相邻城市,已经建有水利设施。由于第N 行的城市靠近沙漠,是该国的干旱区,所以要求其中的每座城市都建有水利设施。那么,这个要求能否满足呢?如果能,请计算最少建造几个蓄水厂;如果不能,求干旱区中不可能建有水利设施的城市数目。
Input
输入文件的每行中两个数之间用一个空格隔开。输入的第一行是两个正整数N 和M,表示矩形的规模。接下来N 行,每行M 个正整数,依次代表每座城市的海拔高度。
Output
输出有两行。如果能满足要求,输出的第一行是整数1,第二行是一个整数,代表最少建造几个蓄水厂;如果不能满足要求,输出的第一行是整数0,第二行是一个整数,代表有几座干旱区中的城市不可能建有水利设施。
Sample Input
2 5
9 1 5 4 3
8 7 6 1 2
Sample Output
1
1
Http
Luogu:https://www.luogu.org/problem/show?pid=1514
Source
搜索,动态规划
解决思路
首先要找到一个顶部的城市能够最多流向底部的哪些城市。如果有解的话,这个底部的城市一定是一个区间。首先证明这一点
如果顶部蓝色城市只能覆盖黄色的两部分
如果有解的话,那么中间的红色部分由这个绿色城市来覆盖
那既然绿色能够覆盖红色,蓝色为什么就不可以呢?可以发现箭头有交叉。
所以我们发现,如果有解的话,顶部的一个城市一定对应一个底部城市的区间,否则一定无解。
那么第一步就是从每一个顶部城市出发,向下一直走到最后一排寻找出那个区间,这里使用bfs实现。这里需要注意的是,如果每一个顶部城市都进行一遍bfs,势必会超时,所以一个剪枝就是如果这个城市的左边或右边海拔比它高,则这个城市不需要bfs,也就是说我们只对那些比它两边都高的城市bfs。因为海拔高的城市在bfs时一定会覆盖海拔低的相邻城市。
那么接下来,问题就转化为给出若干个区间,选择其中最少的覆盖整个区间。
笔者一开始采用的是排序贪心,然后用差分法标记,贪心地选择,但无奈一直没有调对,只好采用动态规划的方法。
具体是设F[i]表示从左向右覆盖到城市i时所用的最少区间数,那么对于一个区间[l,r]而言,有\(j\in [l,r],F[j]=min(F[j],F[l-1]+1)\)
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxN=600;
const int F1[5]={0,1,-1,0,0};//两个F数组分别是在bfs搜索时的四个方向
const int F2[5]={0,0,0,1,-1};
const int inf=2147483647;
class Range//区间
{
public:
int l,r;
};
class Pos//位置,bfs里要用
{
public:
int x,y;
};
bool operator <(Range A,Range B)//之前贪心算法时要用到排序
{
return (A.l<B.l)||((A.l==B.l)&&(A.r>B.r));
}
int n,m;
int Rangecnt=0;//统计总共找到了多少个可行区间
int Height[maxN][maxN];//海拔
bool vis[maxN][maxN];//bfs中标记是否走过
bool is_cover[maxN];//标记最后一排城市是否走过,方便判断无解
Range R[maxN];//区间
queue<Pos> Q;//bfs队列
int F[maxN];//最后求最少区间覆盖时动态规划
void bfs(int st);
int main()
{
memset(is_cover,0,sizeof(is_cover));//初始化
Rangecnt=0;
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)//读入海拔
scanf("%d",&Height[i][j]);
Height[1][0]=0;//为了方便判断,在第一个城市左边和第二个城市右边加入高度为0的城市
Height[1][m+1]=0;
for (int i=1;i<=m;i++)//对于第一行中满足比它两边都高的城市进行bfs搜索区间
if ((Height[1][i]>=Height[1][i-1])&&(Height[1][i]>=Height[1][i+1]))
bfs(i);
int sum=0;//sum用来统计总共覆盖了多少城市,如果sum!=m,则无解
for (int i=1;i<=m;i++)
sum+=(int)(is_cover[i]);
if (sum<m)
{
printf("0\n%d\n",m-sum);
return 0;
}
sort(&R[1],&R[Rangecnt+1]);//其实这个排序是不必要的
memset(F,126,sizeof(F));
F[0]=0;
for (int i=1;i<=Rangecnt;i++)//动态规划求解最少区间覆盖
for (int j=R[i].l;j<=R[i].r;j++)
F[j]=min(F[j],F[R[i].l-1]+1);
printf("1\n%d\n",F[m]);
return 0;
}
void bfs(int st)//bfs求解
{
memset(vis,0,sizeof(vis));
Pos init;
init.x=1;
init.y=st;
vis[1][st]=1;
while (!Q.empty())
Q.pop();
Q.push(init);
do
{
Pos u=Q.front();
Q.pop();
for (int i=1;i<=4;i++)
{
int x=u.x+F1[i];
int y=u.y+F2[i];
if ((x<=n)&&(x>=1)&&(y<=m)&&(y>=1)&&(Height[u.x][u.y]>Height[x][y])&&(vis[x][y]==0))
{
Q.push((Pos){x,y});
vis[x][y]=1;
}
}
}
while (!Q.empty());
int l=-1,r=-1;//记录覆盖区间
for (int i=1;i<=m;i++)
{
if ((vis[n][i]==1)&&(l==-1))
l=i;
if ((vis[n][i]==0)&&(l!=-1)&&(r==-1))
r=i-1;
if (vis[n][i]==1)//同时更新is_cover数组
is_cover[i]=1;
}
if (l==-1)//可能出现当前选择的最上面一排的城市无法到达最下面一排,此时直接退出
return;
if (r==-1)//当r==-1说明覆盖到了最后一个城市
r=m;
Rangecnt++;
R[Rangecnt].l=l;
R[Rangecnt].r=r;
return;
}