nutui-uniapp/components/list/type.ts

京东风格的轻量级移动端 Uniapp、Vue3 组件库（支持小程序开发）

export interface CachedPosition {
  index: number
  top: number
  bottom: number
  height: number
  dValue: number
}

export enum CompareResult {
  eq = 1,
  lt,
  gt,
}

export function binarySearch<T, VT>(list: T[], value: VT, compareFunc: (current: T, value: VT) => CompareResult) {
  let start = 0
  let end = list.length - 1
  let tempIndex: number | null = null

  while (start <= end) {
    tempIndex = Math.floor((start + end) / 2)
    const midValue = list[tempIndex]

    const compareRes: CompareResult = compareFunc(midValue, value)
    if (compareRes === CompareResult.eq)
      return tempIndex

    if (compareRes === CompareResult.lt)
      start = tempIndex + 1

    else if (compareRes === CompareResult.gt)
      end = tempIndex - 1
  }

  return tempIndex!
}