golang算法-计算10000个数里最小的10个数

前言

从包含M个数字的池子里，取出前k个(最小的)数字。

分析

使用mapreduce思想，将M个数字的池，拆成容量为vol的子池，对子池取出最小的10个数，将所有子池的最小十个数合并，再取一下最小10个数。

• 生成M个数字的大数组

func GenMNumberArr(M int, seed time.Time) []int {
  var rs = make([]int, 0, M)
  rand.Seed(seed.UnixNano())
  for i := 0; i < M; i++ {
    rs = append(rs, rand.Intn(1029381))
  }
  return rs
}

// 生成了长度20的随机数组
// 输出[...], 20
func TestGenMNumberArr(t *testing.T) {
  rs := GenMNumberArr(20, time.Now())
  fmt.Println(len(rs))
  fmt.Println(rs)
}

• 子池拆分

func SplitArr(arr []int, vol int) [][]int {
  var rs = make([][]int, 0, 100)
  var tmp []int
L:
  for i, _ := range arr {
    if (i+1)%vol == 0 {
      tmp = arr[0 : i+1]
      rs = append(rs, tmp)
      arr = arr[i+1:]
      goto L
    } else {
      if i == len(arr)-1 {
        rs = append(rs, arr)
      }
      continue
    }
  }
  return rs
}
// 输出 [[1 24 1 2 3] [4 11 12 19 11] [17 19 22 23]]
func TestSplitArr(t *testing.T) {
  rs := SplitArr([]int{1, 24, 1, 2, 3, 4, 11, 12, 19, 11, 17, 19, 22, 23}, 5)
  fmt.Println(rs)
}

• 对包含大于等于K个数字的数组，进行取最小的十个数字的操作

// 从数组中，选出最小的K个数据,并升序排列成数组
func GetMinK(k int, src []int) []int {
  if k > len(src) {
    panic(fmt.Errorf("min k must smaller than src length, but got k '%d', src.len '%d'", k, len(src)))
  }
  var tmp int
  for i := 0; i < k; i++ {
    for j := i + 1; j < len(src); j++ {
      if src[i] > src[j] {
        tmp = src[i]
        src[i] = src[j]
        src[j] = tmp
      }
    }
  }
  return src[:k]
}

// 从数组中，选取了最小的4个
// 输出 [1 2 3 3] 
func TestGetMinK(t *testing.T) {
  rs := GetMinK(4, []int{1, 3, 4, 5, 6, 12, 3, 2, 15, 199})
  fmt.Println(rs)
}

• 最后，从10000个数中，取最小的10个数

func TestGetMin10NumberFrom10000(t *testing.T) {
  arr10000 := GenMNumberArr(10000, time.Now())

  subArrs := SplitArr(arr10000, 100)

  minArr := make([]int, 0, 100)
  for i, _ := range subArrs {
    minArr = append(minArr, GetMinK(10, subArrs[i])...)
  }

  rs :=GetMinK(10, minArr)
  fmt.Println(rs)
}

实现仓库

​​点这里​​