1. 程式人生 > 程式設計 >golang遍歷時修改被遍歷物件的示例詳解

golang遍歷時修改被遍歷物件的示例詳解

前言

很多時候需要將遍歷物件中去掉某些元素,或者往遍歷物件中新增元素,這時候就需要小心操作了。
對於go語言中的一些注意事項我做了總結和示例,留下點筆記。

遍歷切片

1.遍歷切片時去掉元素,錯誤示例:

func main() {
  arr := []int{1,2,3,4}
  for i := range arr {
    if arr[i] == 3 {
      arr = append(arr[:i],arr[i+1:]...)
    }
  }
  fmt.Println(arr)
}

最終報錯panic: runtime error: index out of range [3] with length 3

,因為range在迭代時已經確定i的範圍為[0,len(arr))的左閉右開的區間。
但是當滿足arr[i] == 3時對arr進行了修改,縮短了arr的長度,此時len(arr)=3,最大下標為2,因此當執行arr[3]時會報錯。

2.遍歷切片時去掉元素,不會報錯,但不建議的寫法:

func main() {
  arr := []int{1,4}
  for i,v := range arr {
    fmt.Println(i,v)
    if v == 3 {
      arr = append(arr[:i],arr[i+1:]...)
      // arr = []int{4,5,6,7} // 可以將上一行程式碼替換,看最終遍歷的i,v情況
    }
  }
  fmt.Println(arr)
}

還是回到range的用法,當執行for迴圈時就已經確定(i,v)的遍歷元素值,及時迴圈過程中修改arr,也不會改變for要遍歷的(i,v)值。
可以將上面程式碼修改一下,看下在迴圈中改變arr值時,後面遍歷的(i,v)是不會隨著arr的改變而改變的。

3.遍歷切片時去掉元素,建議寫法:

func main() {
  arr := []int{1,4}
  for i := 0; i < len(arr); i++ {
    fmt.Println(i,arr[i])
    if arr[i] == 3 {
      arr = append(arr[:i],arr[i+1:]...)
      i--
    }
  }
  fmt.Println(arr)
}

該方案只修改i的值,在刪除元素時進行i--,可以確保遍歷arr沒有問題,而且每次通過arr[i]獲取切片值不存在問題。
當然用該方式也可以在遍歷時新增元素,只要i也對應變化就沒問題。

遍歷map

1.遍歷map時去掉元素,可參考官方示例,可看下官方描述,下面這種方案是安全的。

for key := range m {
  if key.expired() {
    delete(m,key)
  }
}

2.清空map所有元素,如下第一種是省事的寫法,第二種不會產生新的物件,用哪種看個人喜好吧。

m = make(map[int]int) // 可以產生一個新物件,舊物件等著被垃圾回收
 
for k := range m {
  delete(m,k) // 迴圈遍歷並刪除map所有元素,好處是map快取還在,下次新增時可直接使用快取
}

總結

關於切片遍歷時進行操作需要注意一些坑。
關於map遍歷時進行操作相對坑少點,不過遍歷map需要修改元素時,map的value要為指標型別,這點得謹記。

到此這篇關於golang遍歷時修改被遍歷物件的文章就介紹到這了,更多相關golang遍歷物件內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!