题目
攀登者喜欢寻找各种地图,并且尝试攀登到最高的山峰。
地图表示为一维数组,数组的索引代表水平位置,数组的元素代表相对海拔高度。其中数组元素0代表地面。
例如:[0,1,2,4,3, 1,0,0,1,2,3,1,2,1,0],代表如下图所示的地图,地图中有两个山脉位置分别为1,2,3,4,5 和8,9,10,11,12,13,最高峰高度
分别为 4,3。最高峰位置分别为3,10。
一个山脉可能有多座山峰(高度大于相邻位置的高度,或在地图边界且高度大于相邻的高度)。
登山者想要知道一张地图中有多少座山峰。
输入描述
输入为一个整型数组,数组长度Q大于1.
输出描述
输出地图中山峰的数量。
用例
输入
0,1,4,3,1,0,0,1,2,3,1,2,1,0
输出
3
def count_peaks(map_array):
num_peaks = 0
length = len(map_array)
# 检查第一个位置
if length > 1 and map_array[0] > map_array[1]:
num_peaks += 1
# 检查中间位置
for i in range(1, length - 1):
if map_array[i] > map_array[i - 1] and map_array[i] > map_array[i + 1]:
num_peaks += 1
# 检查最后一个位置
if length > 1 and map_array[-1] > map_array[-2]:
num_peaks += 1
return num_peaks
# 输入地图数组
input_map = [0, 1, 4, 3, 1, 0, 0, 1, 2, 3, 1, 2, 1, 0]
# 计算山峰数量
num_peaks = count_peaks(input_map)
# 输出结果
print(num_peaks)