go語言學(xué)習(xí)(5)--數(shù)組與切片

數(shù)組的聲明

func array() {
    var arr1 [5]int
    arr2 := [3]int{1, 3, 5}
    // 自行推斷數(shù)組長度
    arr3 := [...]int{1, 3, 5, 7, 9}
    // 二維數(shù)組
    var grid [4][5]int
    fmt.Println(arr1, arr2, arr3, grid)
    // [0 0 0 0 0] [1 3 5] [1 3 5 7 9] [[0 0 0 0 0] [0 0 0 0 0] [0 0 0 0 0] [0 0 0 0 0]]
}

變量的聲明已經(jīng)講過啦,不熟悉的可以看第二章

數(shù)組的遍歷

func arrRange() {
    arr1 := [...]int{2, 4, 6, 8, 10}
    for i := 0; i < len(arr1); i++ {
        fmt.Println(arr1[i])
    }

    for _, value := range arr1 {
        fmt.Println(value)
    }

兩種方法,一種傳統(tǒng)的下標(biāo)遍歷一種上一章講到的range
推薦遍歷用 range, 很方便
在idea里,傳統(tǒng)的 for 循環(huán)智能提示是 for,forr就可以自動(dòng)補(bǔ)全 range了

值傳遞 or 引用傳遞

上一章講到這個(gè)概念了,這一章繼續(xù)舉個(gè)??

func printArray(arr [5]int) {
    arr[0] = 100
    for _, v := range arr {
        println(v)
    }
}

func main() {
    arr1 := [5]int{1, 2, 3, 4, 5}
    arr2 := [3]int{1, 2, 3}
    printArray(arr1)
    printArray(arr2)
}

這個(gè)結(jié)果是什么呢

# command-line-arguments
lesson4(array)/array.go:45:12: cannot use arr2 (type [3]int) as type [5]int in argument to printArray

說的是類型不匹配,我需要一個(gè)[5]int的,你給我傳的是[3]int的,當(dāng)然報(bào)錯(cuò)啦

func printArray(arr [5]int) {
    arr[0] = 100
    for _, v := range arr {
        println(v)
    }
}
func main() {
    arr1 := [5]int{1, 2, 3, 4, 5}
    printArray(arr1)
    fmt.Println(arr1)
}

打印的結(jié)果是什么?
當(dāng)然還是[1,2,3,4,5]啦,因?yàn)槟闶前?arr1拷貝了一份,函數(shù)體內(nèi)的變化并不影響外部的變量
如果想改變 arr[0]的值你得這樣

func printArray2(arr *[5]int) {
    arr[0] = 100
    for _, v := range arr {
        println(v)
    }
}

func main() {
    arr1 := [5]int{1, 2, 3, 4, 5}
    printArray(arr1)
    printArray2(&arr1)
    fmt.Println(arr1)
}

傳一個(gè)指向 arr1的指針,懂了嗎,在go里只有值傳遞,而且 array 這種類型也是值,而在其他語言里比如python,javascript array 都是引用類型
這幾段代碼可能讓你覺得go里的數(shù)組很難用,其實(shí)不然,下面我們講下數(shù)組的切片

切片slice

func main() {
    arr := [...]int{0, 1, 2, 3, 4, 5, 6, 7}
    fmt.Println("arr[2:6] = ", arr[2:6])
    fmt.Println("arr[2:6] = ", arr[:6])
    fmt.Println("arr[2:6] = ", arr[2:])
    fmt.Println("arr[2:6] = ", arr[:])
}

有編程基礎(chǔ)的應(yīng)該都能知道,結(jié)果是:

arr[2:6] =  [2 3 4 5]
arr[2:6] =  [0 1 2 3 4 5]
arr[2:6] =  [2 3 4 5 6 7]
arr[2:6] =  [0 1 2 3 4 5 6 7]

再來看看切片

func updateSlice(s []int) {
    s[0] = 100
}

func main() {
    arr := [...]int{0, 1, 2, 3, 4, 5, 6, 7}
    s1 := arr[2:]
    s2 := arr[:]
    fmt.Println(s2) // [0 1 2 3 4 5 6 7]
    fmt.Println("After updateSlice")
    updateSlice(s1)
    fmt.Println(s1) // [100 3 4 5 6 7]
    fmt.Println(arr) // [0 1 100 3 4 5 6 7]
}

Slice 本身沒有數(shù)據(jù),只是對(duì)底層 array 的一個(gè) view
鞏固一下

// s2 = [0 1 100 3 4 5 6 7]
fmt.Println("reSlice")
s2 = s2[:5]
fmt.Println(s2) // [0 1 100 3 4]
s2 = s2[2:]
fmt.Println(s2) // [100 3 4]

這些都理解,下面再看個(gè)例子

arr = [...]int{0,1,2,3,4,5,6,7}
s1 = arr[2:6]
s2 = s1[3:6]
fmt.Println("s1 = ",s1)
fmt.Println("s2 = ",s2)

這是不是有問題呢? s1是 arr 的第3個(gè)到第6個(gè)元素,總共4個(gè)
s2是不是下標(biāo)越界了呢
運(yùn)行一下

s1 =  [2 3 4 5]
s2 =  [5 6 7]

再來看看
fmt.Println(s1[4])
panic: runtime error: index out of range
很奇怪對(duì)不對(duì),奇怪就對(duì)了
記住這個(gè)概念
Slice 本身沒有數(shù)據(jù),只是對(duì)底層 array 的一個(gè) view
來看個(gè)圖

image

s1為[2 3 4 5],s2為[5 6 7]
slice 可以向后擴(kuò)展,不可以向前擴(kuò)展
s[i]不可以超越len(s),向后擴(kuò)展不可以超越cap(s)
這就是cap capacity(容量)的概念

fmt.Printf("s1=%v,len(s1)=%d,cap(s1)=%d\n",
        s1, len(s1), cap(s1))
fmt.Printf("s2=%v,len(s2)=%d,cap(s2)=%d\n",
        s2, len(s2), cap(s2))

// s1=[2 3 4 5],len(s1)=4,cap(s1)=6
// s2=[5 6 7],len(s2)=3,cap(s2)=3

go的數(shù)組可以指定起始,結(jié)束和 capacity,而python是指定起始,結(jié)束和步長,不要混為一談
s1[1:2:3]指定 cap 為3
slice起初我也覺得有點(diǎn)繞,多練習(xí)就好了

append

arr := [...]int{0, 1, 2, 3, 4, 5, 6, 7}
s1 := arr[2:6]
s2 := s1[3:5]
s3 := append(s2, 10)
s4 := append(s3, 11)
s5 := append(s4, 12)
fmt.Println(s1, s2, s3, s4, s5)

// [2 3 4 5] [5 6] [5 6 10] [5 6 10 11] [5 6 10 11 12]

append和其他語言一樣,可以突破capacity的限制,依次在數(shù)組后面添加元素

擴(kuò)容的規(guī)律

func printSlice(s []int) {
    fmt.Printf("len=%d, cap=%d\n", len(s), cap(s))
}

var s []int
for i := 0; i < 100; i++ {
    printSlice(s)
    s = append(s, 2*i+1)
}

貼幾個(gè)結(jié)果感受下

len=14, cap=16
len=15, cap=16
len=16, cap=16
...
len=31, cap=32
len=32, cap=32
...
len=63, cap=64
len=64, cap=64
len=65, cap=128
len=66, cap=128

和java不同,java默認(rèn)負(fù)載因子是0.75,go看的出來是capacity滿了才擴(kuò)容,每次擴(kuò)容兩倍,所以和java一樣,數(shù)組最好知道容量,上來就建好

還有一種創(chuàng)建數(shù)組的方法make

s2 = make([]int, 16)
s3 = make([]int, 10, 32)
printSlice(s2)
printSlice(s3)

// len=16, cap=16
// len=10, cap=32

后面會(huì)經(jīng)常用到,創(chuàng)建數(shù)組和channel都很常用
copy(s2, s1)

func copy(dst, src []Type) int

fmt.Println(s2)

// [2 3 4 5 0 0 0 0 0 0 0 0 0 0 0 0]

刪除數(shù)組元素

fmt.Println("Deleting elements from slice")
s2 = append(s2[:3], s2[4:]...)
printSlice(s2)

fmt.Println("Poping from front")
front := s2[0]
s2 = s2[1:]
fmt.Println(front)
printSlice(s2)

fmt.Println("Poping from tail")
tail := s2[len(s2)-1]
s2 = s2[:len(s2)-1]
fmt.Println(tail)
printSlice(s2)

/**
Deleting elements from slice
len=15, cap=16
Poping from front
2
len=14, cap=15
Poping from tail
0
len=13, cap=15
*/

總結(jié)

go里只有值傳遞
不要直接用 array, 要用切片來操作
注意cap 這個(gè)隱含的值,你如果不想別人能訪問到切片以外的數(shù)據(jù),可以加上 cap 比如s[1:2:3]
切片只是對(duì)底層 array 的 view

上述代碼均已上傳至 github, 歡迎 star
https://github.com/yejunyu/golearn