Board logo

标题: 挑出峰值数字和所处位置(coderwars 5级挑战题) [打印本页]

作者: ivor    时间: 2020-5-29 17:54     标题: 挑出峰值数字和所处位置(coderwars 5级挑战题)

本帖最后由 ivor 于 2020-5-29 21:53 编辑

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


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

详细介绍:

例子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    时间: 2020-5-29 19:20

这个峰值有什么范围吗?
作者: ivor    时间: 2020-5-29 20:09

回复 2# Gin_Q


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

形象比喻就像山头。
作者: Gin_Q    时间: 2020-5-29 21:13

那第一位和最后一位是不是漏掉了?
作者: xczxczxcz    时间: 2020-5-31 19:55

严格说,前后都要算,
补个 ps
  1. $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;
  2. Function Get-SpaceValue {
  3. param(
  4. [Parameter(Mandatory = $true)][double[]]$arr,
  5. [ValidateSet(0, -1)][sbyte]$y = -1 #[0]不计算两端 [-1]计算两端
  6. )
  7. $r = for ($i = $y; $i -lt $arr.Count - 2 - $y; $i++) {
  8. if ([Linq.Enumerable]::Max([double[]]$arr[$i..($i + 2)]) -eq $arr[$i + 1]) {
  9. @{p = $i + 1; v = $arr[$i + 1] }
  10. }
  11. }
  12. return ConvertTo-Json -Input @{位置 = $r.p; 峰值 = $r.v } -Comp
  13. }
  14. 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    时间: 2020-6-1 08:03

回复 5# xczxczxcz


    忽略了一种相邻两个(或更多个)数可能是相等的情况。
作者: xczxczxcz    时间: 2020-6-1 09:36

回复 6# ivor


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




欢迎光临 批处理之家 (http://www.bathome.net/) Powered by Discuz! 7.2