P1514 引水入城

【題目描述】
見鏈接

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;
}