题目
题目描述:
在一个遥远的国度,一侧是风景秀美的湖泊,另一侧则是漫无边际的沙漠。该国的行政区划十分特殊,刚好构成一个 N 行 M 列的矩形,其中每个格子都代表一座城市,每座城市都有一个海拔高度。
为了使居民们都尽可能饮用到清澈的湖水,现在要在某些城市建造水利设施。水利设施有两种,分别为蓄水厂和输水站。蓄水厂的功能是利用水泵将湖泊中的水抽取到所在城市的蓄水池中。
因此,只有与湖泊毗邻的第 1 行的城市可以建造蓄水厂。而输水站的功能则是通过输水管线利用高度落差,将湖水从高处向低处输送。故一座城市能建造输水站的前提,是存在比它海拔更高且拥有公共边的相邻城市,已经建有水利设施。由于第 N 行的城市靠近沙漠,是该国的干旱区,所以要求其中的每座城市都建有水利设施。那么,这个要求能否满足呢?如果能,请计算最少建造几个蓄水厂;如果不能,求干旱区中不可能建有水利设施的城市数目。
输入格式:
第一行两个整数 n 和 m,
接下来 n 行每行 m 个整数,表示海拔高度。
输出格式:
如果所有干旱区都可以得到水,则第一行输出 1,第二行输出最少的蓄水厂数量。
否则第一行输出 0,第二行输出有几个城市不可能有水。
样例输入:
2 5
9 1 5 4 3
8 7 6 1 2
样例输出:
1
1
样例输入:
3 6
8 4 5 6 4 4
7 3 4 3 3 3
3 2 2 1 1 2
样例输出:
1
3
数据规模:
n≥2,n,m≤500,海拔高度不超过10^6。
代码(WA 80)
#include <bits/stdc++.h>
const int N = 510;
const int dir[][2] = {0, 1, 1, 0, 0, -1, -1, 0};
bool vis[N][N];
int n, m, ans, a[N][N];
void input() {
std::cin >> n >> m;
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
std::cin >> a[i][j];
}
void dfs(int x, int y) {
vis[x][y] = true;
for (int i = 0; i < 4; ++i) {
int fx = x + dir[i][0], fy = y + dir[i][1];
if (fx < 1 || fx > n || fy < 1 || fy > m) continue;
if (vis[fx][fy]) continue;
if (a[fx][fy] >= a[x][y]) continue;
dfs(fx, fy);
}
}
void work() {
while (true) {
int cnt = 0;
for (int i = 1; i <= m; ++i)
if (!vis[n][i]) ++cnt;
if (cnt == 0) {
std::cout << "1\n" << ans << std::endl;
return ;
}
std::vector<int> id;
for (int i = 1; i <= m; ++i)
if (!vis[1][i]) id.push_back(i);
if (id.empty()) {
std::cout << "0\n" << cnt << std::endl;
return ;
}
std::sort(id.begin(), id.end(), [](int x, int y){
return a[1][x] > a[1][y];
});
++ans;
dfs(1, id[0]);
}
}
int main() {
input();
work();
return 0;
}