当前位置: 代码迷 >> 综合 >> 1106. 山峰和山谷
  详细解决方案

1106. 山峰和山谷

热度:18   发布时间:2024-01-17 14:04:24.0

FGD小朋友特别喜欢爬山,在爬山的时候他就在研究山峰和山谷。

为了能够对旅程有一个安排,他想知道山峰和山谷的数量。

给定一个地图,为FGD想要旅行的区域,地图被分为 n×n 的网格,每个格子 (i,j) 的高度 w(i,j) 是给定的。

若两个格子有公共顶点,那么它们就是相邻的格子,如与 (i,j) 相邻的格子有(i?1,j?1),(i?1,j),(i?1,j+1),(i,j?1),(i,j+1),(i+1,j?1),(i+1,j),(i+1,j+1)。

我们定义一个格子的集合 S 为山峰(山谷)当且仅当:

S 的所有格子都有相同的高度。
S 的所有格子都连通。
对于 s 属于 S,与 s 相邻的 s′ 不属于 S,都有 ws>ws′(山峰),或者 ws<ws′(山谷)。
如果周围不存在相邻区域,则同时将其视为山峰和山谷。
你的任务是,对于给定的地图,求出山峰和山谷的数量,如果所有格子都有相同的高度,那么整个地图即是山峰,又是山谷。

输入格式

第一行包含一个正整数 n,表示地图的大小。

接下来一个 n×n 的矩阵,表示地图上每个格子的高度 w。

输出格式

共一行,包含两个整数,表示山峰和山谷的数量。

数据范围

1≤n≤1000,
0≤w≤109

输入样例1:
5
8 8 8 7 7
7 7 8 8 7
7 7 7 7 7
7 8 8 7 8
7 8 8 8 8
输出样例1:
2 1
输入样例2:
5
5 7 8 3 1
5 5 7 6 6
6 6 6 2 8
5 7 2 5 8
7 1 0 1 7
输出样例2:
3 3

样例解释
样例1:

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-hut1hyG6-1648952536434)(1106.%20%E5%B1%B1%E5%B3%B0%E5%92%8C%E5%B1%B1%E8%B0%B7.assets/19_0250799aef-1.png)]

样例2:

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-x94JbWml-1648952536436)(1106.%20%E5%B1%B1%E5%B3%B0%E5%92%8C%E5%B1%B1%E8%B0%B7.assets/19_08db5e60ef-2.png)]

代码:
/*
找到高度一致的连通块,若该连通块周围没有存在比它高的则该连通块叫山峰没有存在比它矮的则该连通块叫山谷
*/
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int N = 1e3 + 10;
int n;
int w[N][N];
bool st[N][N];void bfs(int sx, int sy, bool &has_higher, bool &has_lower)
{queue<PII> qu;st[sx][sy] = 1;qu.push({sx, sy});while (qu.size()){PII t = qu.front();qu.pop();for (int i = t.first - 1; i <= t.first + 1; i++){for (int j = t.second - 1; j <= t.second + 1; j++){if (i == t.first && j == t.second)continue;if (i < 1 || i > n || j < 1 || j > n)continue;if (w[i][j] != w[t.first][t.second]){if (w[i][j] > w[t.first][t.second])has_lower = true;elsehas_higher = true;}else if (!st[i][j]){qu.push({i, j});st[i][j] = 1;}}}}
}
int main()
{cin >> n;for (int i = 1; i <= n; i++)for (int j = 1; j <= n; j++)cin >> w[i][j];int peak = 0, valley = 0;for (int i = 1; i <= n; i++){for (int j = 1; j <= n; j++){if (!st[i][j]){bool has_higher = false, has_lower = false;bfs(i, j, has_higher, has_lower);if (!has_higher)peak++;if (!has_lower)valley++;}}}cout << valley << " " << peak;return 0;
}