洛谷 P1514 引水入城

题目描述

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

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

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

【输入样例1】

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

【输入样例2】

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

输出样例#1：

【输出样例1】

1
1

【输出样例2】

1
3
说明

【样例1 说明】

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

【样例2 说明】

上图中，在3 个粗线框出的城市中建造蓄水厂，可以满足要求。以这3 个蓄水厂为源头

在干旱区中建造的输水站分别用3 种颜色标出。当然，建造方法可能不唯一。

【数据范围】

分析

开始看到这道题时，第一反应就是搜索。认真分析题目后发现，对问题的求解并非无规律可循。通过对样例的观察，我猜想：对于第一行的每一个格子，它所能影响到的最后一行的格子会不会是一段区间呢？
在能满足条件的情况下，答案是肯定的。下面是我对这一问题的证明：
假设第一行的A格所能影响的最后一行的区间为[l1,r1]和[l2,r2]，r1

代码

#include<cstdio>
#include<cstring>
#include<algorithm>

#define N 505

using namespace std;

int i,j,n,m,ans;
int a[N][N];
bool exist[N][N];
bool judge[N][N];

struct NOTE
{
    int a,b;
}s[501];

void dfs(int x, int y)
{
    if (exist[x][y]) 
        return;
    exist[x][y]=1;
    if (x+1<=n&&a[x][y]>a[x+1][y]) 
        dfs(x+1,y);
    if (x-1>=1&&a[x][y]>a[x-1][y]) 
        dfs(x-1,y);
    if (y+1<=m&&a[x][y]>a[x][y+1]) 
        dfs(x,y+1);
    if (y-1>=1&&a[x][y]>a[x][y-1]) 
        dfs(x,y-1);
}

int comp(NOTE a, NOTE b)
{
    return a.a<b.a;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (i=1; i<=n; i++)
        for (j=1; j<=m; j++)
            scanf("%d",&a[i][j]);
    for (i=1; i<=m; i++)
    {
        memset(exist,0,sizeof(exist));
        dfs(1,i);
        for (j=1; j<=m; j++)
            judge[i][j]=exist[n][j];
    }
    int cnt=m;
    for (i=1; i<=m; i++)
        for (j=1; j<=m; j++)
            if (judge[j][i])
            {
                cnt--;
                break;
            }
    if (cnt>0)
    {
        printf("0\n%d",cnt);
        return 0;
    }
    for (i=1; i<=m; i++)
    {
        for (j=1; j<=m; j++) 
            if (judge[i][j]) 
            break;
        s[i].a=j;
        for (j=s[i].a+1; j<=m; j++)
            if (judge[i][j]==0) 
                break;
        s[i].b=j-1;
    }
    sort(s+1,s+m+1,comp);
    int maxv=0,tmp=0;
    for (i=1; i<=m; i++) 
    {
        if (s[i].a>m) 
            break;
        if (s[i].a<=1+maxv) 
            tmp=max(tmp,s[i].b);
        else
        {
            maxv=tmp;
            ans++;
            tmp=max(s[i].b,tmp);
        }      
    }
    if (maxv!=m) 
        ans++;
    printf("1\n%d",ans);
    return 0;
}