矢量搜索算法

Question

我有以下问题。 说我有一个矢量：

v = [1,2,3,4,5,1,2,3,4,...]

我想从矢量中顺序采样点，其中绝对的maginute差异高于先前采样点的阈值。 所以说我的门槛是2。

我从索引1开始，然后对第一点进行采样1.然后在v [3]处满足我的条件，并且我采样3（因为3-1> = 2）。 然后3，新的采样点成为参考，我检查。 下一个采样点是5，即v [5]（5-3> = 2）。 然后下一个点是1，即v [6]（abs（1-5）> = 2）。

不幸的是，我在R中的代码花了太长时间。 基本上我反复扫描阵列并寻找匹配。 我认为这种方法很天真。 我有一种感觉，我可以通过数组一次完成这项任务。 我不知道怎么回事。 任何帮助赞赏。 我想我遇到的问题是下一个采样点的位置可以是数组中的任何位置，我需要扫描数组从当前点到结束才能找到它。

谢谢。

Answer 1

我没有看到没有循环可以做到这一点的方法，所以这里有一个：

my.sample <- function(x, thresh) {
   out <- x
   i <- 1
   for (j in seq_along(x)[-1]) {
      if (abs(x[i]-x[j]) >= thresh) {
         i <- j
      } else {
         out[j] <- NA
      }
   }
   out[!is.na(out)]
}

my.sample(x = c(1:5,1:4), thresh = 2)
# [1] 1 3 5 1 3

Answer 2

你可以在没有循环的情况下使用一点递归来做到这一点：

vsearch = function(v, x, fun=NULL) {
  # v: input vector
  # x: threshold level

  if (!length(v) > 0) return(NULL)

  y = v-rep(v[1], times=length(v))
  if (!is.null(fun)) y = fun(y)

  i = which(y >= x)

  if (!length(i) > 0) return(NULL)
  i = i[1]

  return(c(v[i], vsearch(v[-(1:(i-1))], x, fun=fun)))
}

使用上面的矢量：

> vsearch(c(1,2,3,4,5,1,2,3,4), 2, abs)
[1] 3 5 1 3