【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 种颜色标出。当然，建造方法可能不唯一。

【数据范围】

【思路】搜索每一个蓄水厂，得到它在终点能覆盖的区间，能证到若所有终点都能被覆盖，那么这个区间一定是连续的（什么时候有空去分析一下）。再把得到的区间排一下序，用贪心做到解。注意，当前算法有一个点超时。那么就要剪枝。

【剪枝】第一行的点：若它比两边的点都要低，那么它就没有存在的意义，肯定会被刷掉，因此就不用再算了。

【代码】

#include <bits/stdc++.h>
using namespace std;

struct nodest{
    int l,r;
}s[505];
int n,m,a[505][505],b[505][505],c[505];
int dx[4]={1,0,-1,0},dy[4]={0,1,0,-1};

int CMP(nodest a,nodest b)
{
    return a.l<b.l||a.l==b.l&&a.r<b.r;
}

int dfs(int x,int y)
{
    b[x][y]++;
    for(int i=0;i<4;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if(nx<=0||nx>n||ny<=0||ny>m) continue;
        if(a[nx][ny]<a[x][y]&&b[nx][ny]==0)
            dfs(nx,ny);
    }
    return 0;
}

int gdal()
{
    int xx=1,yy=0,ans=0,fst=1;
    while(1)
    {
        int by=0,bf=0;
        for(int i=fst;i<=m;i++)
            if(xx<=s[i].l&&s[i].l<=yy+1&&s[i].r>by)
                by=s[i].r,bf=i;
        yy=by;
        fst=bf;
        ans++;
        if(yy==m) return ans;
    }
}

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++)
    {
        memset(b,0,sizeof(b));
        if(a[1][i]>=a[1][i-1]&&a[1][i]>=a[1][i+1])
		{
			dfs(1,i);
			s[i].l=INT_MAX,s[i].r=INT_MIN;
        	for(int j=1;j<=m;j++)
        		if(b[n][j]) s[i].l=min(s[i].l,j),s[i].r=max(s[i].r,j),c[j]++;
		}
        else
        {
        	s[i].l=0;
        	s[i].r=0;
		}
    }
    
    int cnt=0;
    for(int i=1;i<=m;i++)
        if(c[i]==0) cnt++;
    if(cnt)
    {
        cout<<"0"<<endl<<cnt;
        return 0;
    }
    
    for(int i=1;i<=m;i++)
        if(s[i].l==INT_MAX||s[i].r==INT_MIN)
        {
            s[i].l=0;
            s[i].r=0;
        }
    sort(s+1,s+m+1,CMP);
    
    cout<<"1"<<endl<<gdal();
    
    return 0;
}