【題目描述】
見鏈接
P1514 引水入城
【分析】
bfs填色問題變式,如果從第一行能流到最後一行就保留,統計一下最後覆蓋的區間是否覆蓋滿最後一行;然後運用貪心,選擇連續的且覆蓋最多的蓄水站即可,如果不滿足,輸出缺少的即可。
樣例:
【參考代碼】
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int ans,n,m,a[505][505],dx[5]={0,0,0,1,-1},dy[5]={0,1,-1,0,0},num;
bool f[505][505],ff[505][505];
struct node
{
int x;
int y;
}line [100000];//保存第一行的點能覆蓋的最後一行的區間
struct node1
{
int x;
int y;
}q[250001];
void bfs(int c,int r)//廣搜,洪水填充
{
memset(ff,false,sizeof(ff));
int head,tail,xx,yy,i;
head=0;
tail=1;
ff[c][r]=true;
q[1].x=c;
q[1].y=r;
do
{
++head;
for (i=1;i<=4;++i)
{
xx=q[head].x+dx[i];
yy=q[head].y+dy[i];
if (xx>n||xx<1||yy>m||yy<1)
continue;
if (!ff[xx][yy]&&a[q[head].x][q[head].y]>a[xx][yy])
{
++tail;
q[tail].x=xx;
q[tail].y=yy;
f[xx][yy]=true;
ff[xx][yy]=true;
}
}
}while (head<tail);
for (i=1;i<=m;++i)//判斷第一行該點影響最後一行的區間的左端點。
if (ff[n][i])
{
line[++num].x=i;
break;
}
for (i=m;i>=1;--i)//判斷第一行該點影響最後一行的區間的右端點。
if (ff[n][i])
{
line[num].y=i;
break;
}
if (line[num].x==line[num].y&&line[num].x==0)
--num;//如果無法覆蓋到最後一行就取消這個區間。
}
bool cmp(node a,node b)
{
return a.x<b.x||a.x==b.x&&a.y>b.y;
}
int main()
{
int i,j;
cin>>n>>m;
for (i=1;i<=n;++i)
for (j=1;j<=m;++j)
cin>>a[i][j];
for (j=1;j<=m;++j)
if (!f[1][j]&&a[1][j]>=a[1][j-1]&&a[1][j]>=a[1][j+1])
{
f[1][j]=true;
bfs(1,j);//如果該點之前沒有被填充過就填充。
}
int s=0;
for (i=1;i<=m;++i)
if (f[n][i]==true)
++s;//統計最後一行能被覆蓋的個數
if (s==m)
{
cout<<1<<endl;
sort(line+1,line+num+1,cmp);
int xx=1;
int yy=0;
int s=0;
while (1)
{
int by=0;
for (i=1;i<=num;++i)
if (line[i].x>=xx&&line[i].x<=yy+1&&line[i].y>by)
by=line[i].y;
yy=by;
++s;
if (yy==m)
break;
}
cout<<s;
}
else
cout<<0<<endl<<m-s;
}