快捷導(dǎo)航

解決Golang map range遍歷結(jié)果不穩(wěn)定問(wèn)題

更新時(shí)間：2020年12月23日 10:01:12 作者：Lucus小秋

這篇文章主要介紹了解決Golang map range遍歷結(jié)果不穩(wěn)定問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

閑言少敘，本文主要是想介紹一個(gè)Golang開(kāi)發(fā)常見(jiàn)的一個(gè)問(wèn)題。然而，此問(wèn)題對(duì)于初學(xué)者來(lái)說(shuō)卻經(jīng)常容易陷入坑中。

問(wèn)題

我在寫一段代碼時(shí)，使用了Golang的map數(shù)據(jù)結(jié)構(gòu)，目的是想用map緩存計(jì)數(shù)結(jié)果。簡(jiǎn)單來(lái)說(shuō)map的鍵也是整型的，且以遞增順序存儲(chǔ)。我的最初想法是，在統(tǒng)計(jì)結(jié)束后，按照map中存儲(chǔ)的鍵有序輸出值。可是，當(dāng)我運(yùn)行程序時(shí)，結(jié)果并不是我想要的，而且有一定概率運(yùn)行結(jié)果不同。

問(wèn)題代碼

func sortByBits(arr []int) []int {
  var bitmap = make(map[int][]int, 0)
  // map緩存
  for i := 0; i < len(arr); i++ {
    bits := calBits(arr[i])
    if _, ok := bitmap[bits]; !ok {
      tmp := make([]int, 0)
      tmp = append(tmp, arr[i])
      bitmap[bits] = tmp
    } else {
      bitmap[bits] = append(bitmap[bits], arr[i])
      for j := len(bitmap[bits]) - 1;j > 0; j-- {
        if bitmap[bits][j] < bitmap[bits][j - 1] {
          bitmap[bits][j], bitmap[bits][j - 1] = bitmap[bits][j - 1], bitmap[bits][j]
        }
      }
    }
  }
  // 輸出
  var res []int
  for _, value := range bitmap {
    res = append(res, value...)
  }
  return res
}
func calBits(n int) int {
  sum := 0
  for n > 0 {
    if n & 1 == 1 {
      sum++
    }
    n = n >> 1
  }
  return sum
}

當(dāng)發(fā)現(xiàn)這個(gè)問(wèn)題后，我使用二維數(shù)組替換map，改造代碼如下：

改造代碼

func sortByBits(arr []int) []int {
  var bitmap = make([][]int, 0, 10000)
  // map緩存
  for i := 0; i < len(arr); i++ {
    bits := calBits(arr[i])
    if len(bitmap) <= bits {
      length := bits - len(bitmap) + 1
      for j := 0; j < length; j++ {
        tmp := make([]int, 0)
        bitmap = append(bitmap, tmp)
      }
    }
      
    if len(bitmap[bits]) == 0 {
      bitmap[bits] = append(bitmap[bits], arr[i])
    } else {
      bitmap[bits] = append(bitmap[bits], arr[i])
      for j := len(bitmap[bits]) - 1;j > 0; j-- {
        if bitmap[bits][j] < bitmap[bits][j - 1] {
          bitmap[bits][j], bitmap[bits][j - 1] = bitmap[bits][j - 1], bitmap[bits][j]
        }
      }
    }
  }
  // 輸出
  var res []int
  for _, value := range bitmap {
    res = append(res, value...)
  }
  return res
}
func calBits(n int) int {
  sum := 0
  for n > 0 {
    if n & 1 == 1 {
      sum++
    }
    n = n >> 1
  }
  return sum
}

代碼很簡(jiǎn)單，問(wèn)題也很簡(jiǎn)單。原理就是Golang的map按鍵輸出結(jié)果是隨機(jī)的，這是語(yǔ)言本身的一個(gè)特性，或者說(shuō)是"坑"。作為開(kāi)發(fā)者，要熟練掌握語(yǔ)言最基本特性，才能開(kāi)發(fā)出最穩(wěn)健的程序。

補(bǔ)充：golang if _,ok:=range map; ok判斷key是否在map中

由于golang沒(méi)有提供item是否在array當(dāng)中的判斷方法，如果程序里面頻繁用到了這種判斷，可以將array轉(zhuǎn)化為以array當(dāng)中的成員為key的map再用上面的方法進(jìn)行判斷，這樣會(huì)提高判斷的效率。