挑出峰值数字和所处位置（coderwars 5级挑战题）

ivor

看着还不错，和大家分享一下。


名称：挑出峰值数字和所处位置

详细介绍：

例子1, 数组 arr = [0, 1, 2, 5, 1, 0] 峰值索引位置在 3，峰值对应为 5 (since arr[3] equals 5). 结果：{pos: [3], peaks: [5]}
例子2，数组 arr = [1,2,5,4,3,2,3,6,4,1,2,3,3,4,5,3,2,1,2,3,5,5,4,3]), 索引位置在[2,7,14,20] 峰值对应为[5,6,5,5]  结果 ：{"pos":[2,7,14,20], "peaks":[5,6,5,5]})


抛砖引玉附上Python代码:

1. a = [11, 20, 14, 4, 5, 4, 8, 17, 14, 9, 17, 20, 8, 19, 17, 15, 13, 20, 16, 7, 2, 11, 14, 18, 11, 2, 15, 1, 5, 4, 5, 1, 6, 6, 9, 13, 18, 19, 9, 17, 12, 19, 17, 2, 13, 9, 19, 2, 11]
   
2. def pick_peaks(arr):
   
3.     pos = []
   
4.     prob_peak = False
   
5.     for i in range(1, len(arr)):
   
6.         if arr[i] > arr[i-1]:
   
7.             prob_peak = i
   
8.         elif arr[i] < arr[i-1] and prob_peak:
   
9.             pos.append(prob_peak)
   
10.             prob_peak = False
    
11.     return {'pos':pos, 'peaks':[arr[i] for i in pos]}
    
12. print(pick_peaks(a))

复制代码

Gin_Q

这个峰值有什么范围吗？

ivor

回复 2# Gin_Q


    比如：例子2索引位置14比左右数值都小，如果和右边数值相等，则继续往后找，找到后才能确定14位置为峰值，如果右边找不到更小的数值则不成立。

形象比喻就像山头。

Gin_Q

那第一位和最后一位是不是漏掉了？

xczxczxcz

严格说，前后都要算，
补个 ps

1. 
   
2. $A = 20, 14, 4, 5, 4, 8, 17, 14, 9, 17, 20, 8, 19, 17, 15, 13, 20, 16, 7, 2, 11, 14, 18, 11, 2, 15, 1, 5, 4, 5, 1, 6, 6, 9, 13, 18, 19, 9, 17, 12, 19, 17, 2, 13, 9, 19, 2, 11;
   
3. Function Get-SpaceValue {
   
4.         param(
   
5.                 [Parameter(Mandatory = $true)][double[]]$arr,
   
6.                 [ValidateSet(0, -1)][sbyte]$y = -1 #[0]不计算两端 [-1]计算两端
   
7.         )
   
8. 
   
9.         $r = for ($i = $y; $i -lt $arr.Count - 2 - $y; $i++) {
   
10.                 if ([Linq.Enumerable]::Max([double[]]$arr[$i..($i + 2)]) -eq $arr[$i + 1]) {
    
11.                         @{p = $i + 1; v = $arr[$i + 1] }
    
12.                 }
    
13.         }
    
14.         return ConvertTo-Json -Input @{位置 = $r.p; 峰值 = $r.v } -Comp
    
15. }
    
16. Get-SpaceValue -arr $A
    

复制代码

结果

1. {"峰值":[20,5,17,20,19,20,18,15,5,5,6,19,17,19,13,19,11],"位置":[0,3,6,10,12,16,22,25,27,29,31,36,38,40,43,45,47]}

复制代码

ivor

回复 5# xczxczxcz


    忽略了一种相邻两个(或更多个)数可能是相等的情况。

xczxczxcz

回复 6# ivor


    确实忽略了，看来还是挨个比简单。