【Go-6】数据结构与集合

发布于:2025-06-02 ⋅ 阅读:(23) ⋅ 点赞:(0)

6. 数据结构与集合

数据结构是编程中用于组织和存储数据的方式,直接影响程序的效率和性能。Go语言提供了多种内置的数据结构,如数组、切片、Map和结构体,支持不同类型的数据管理和操作。本章将详细介绍Go语言中的主要数据结构与集合,涵盖它们的定义、使用方法、操作技巧以及底层原理。通过丰富的示例和深入的解释,帮助你全面掌握Go语言的数据结构,为构建高效、可维护的程序奠定坚实的基础。

6.1 数组

数组是具有固定大小和相同类型元素的有序集合。在Go语言中,数组的长度是其类型的一部分,这意味着具有不同长度的数组属于不同的类型。

数组的声明与初始化

1. 声明数组

使用var关键字声明数组时,需要指定数组的长度和元素类型。

var arr [5]int

解释:

  • arr是一个长度为5的整型数组。
  • 所有元素默认初始化为0。

2. 声明并初始化数组

可以在声明数组的同时为其元素赋值。

var arr [3]string = [3]string{"apple", "banana", "cherry"}

简化声明:

当声明和初始化数组时,Go可以根据初始化的元素数量自动推断数组的长度。

arr := [3]string{"apple", "banana", "cherry"}

使用省略长度

通过使用...,Go可以根据初始化的元素数量自动确定数组的长度。

arr := [...]float64{1.1, 2.2, 3.3, 4.4}

3. 多维数组

Go支持多维数组,最常见的是二维数组。

var matrix [3][4]int

初始化二维数组:

matrix := [2][3]int{
    {1, 2, 3},
    {4, 5, 6},
}

完整示例:

package main

import "fmt"

func main() {
    // 声明并初始化一维数组
    var arr [5]int = [5]int{1, 2, 3, 4, 5}
    fmt.Println("一维数组:", arr)

    // 使用省略长度声明数组
    arr2 := [...]string{"Go", "Python", "Java"}
    fmt.Println("省略长度的一维数组:", arr2)

    // 声明并初始化二维数组
    matrix := [2][3]int{
        {1, 2, 3},
        {4, 5, 6},
    }
    fmt.Println("二维数组:", matrix)
}

输出:

一维数组: [1 2 3 4 5]
省略长度的一维数组: [Go Python Java]
二维数组: [[1 2 3] [4 5 6]]
数组的操作

1. 访问数组元素

通过索引访问数组元素,索引从0开始。

package main

import "fmt"

func main() {
    arr := [3]string{"apple", "banana", "cherry"}
    fmt.Println("第一个元素:", arr[0]) // 输出: apple
    fmt.Println("第二个元素:", arr[1]) // 输出: banana
    fmt.Println("第三个元素:", arr[2]) // 输出: cherry
}

2. 修改数组元素

数组元素是可修改的,只需通过索引赋值。

package main

import "fmt"

func main() {
    arr := [3]int{10, 20, 30}
    fmt.Println("原数组:", arr)

    arr[1] = 25
    fmt.Println("修改后的数组:", arr) // 输出: [10 25 30]
}

3. 遍历数组

使用for循环或range关键字遍历数组。

使用传统for循环:

package main

import "fmt"

func main() {
    arr := [3]string{"apple", "banana", "cherry"}

    for i := 0; i < len(arr); i++ {
        fmt.Printf("元素 %d: %s\n", i, arr[i])
    }
}

使用range遍历:

package main

import "fmt"

func main() {
    arr := [3]string{"apple", "banana", "cherry"}

    for index, value := range arr {
        fmt.Printf("元素 %d: %s\n", index, value)
    }
}

4. 数组长度

数组的长度是其类型的一部分,可以通过len函数获取。

package main

import "fmt"

func main() {
    arr := [5]int{1, 2, 3, 4, 5}
    fmt.Println("数组长度:", len(arr)) // 输出: 5
}

5. 数组作为函数参数

在Go中,数组作为函数参数时,会复制整个数组。因此,对于大数组,推荐使用指针或切片。

package main

import "fmt"

// 函数接收数组参数
func printArray(arr [3]int) {
    for _, v := range arr {
        fmt.Println(v)
    }
}

func main() {
    arr := [3]int{1, 2, 3}
    printArray(arr)
}

输出:

1
2
3
注意事项
  • 固定长度:数组的长度在声明时固定,无法动态改变。如果需要动态长度,建议使用切片。

  • 类型区别:不同长度的数组属于不同类型,即[3]int[4]int是不同的类型。

    var a [3]int
    var b [4]int
    // a = b // 编译错误: cannot use b (type [4]int) as type [3]int in assignment
    
  • 数组拷贝:数组作为值类型会被复制。因此,在函数中修改数组不会影响原数组,除非使用指针传递。

6.2 切片

切片是基于数组的动态数据结构,比数组更灵活和强大。切片的长度和容量可以动态变化,是Go语言中最常用的数据结构之一。

切片的声明与初始化

1. 声明切片

切片不需要在声明时指定长度,可以通过多种方式声明。

var s []int

解释:

  • s是一个整型切片,初始为nil

2. 使用make函数创建切片

make函数用于创建切片、Map和Channel。对于切片,make需要指定类型、长度和可选的容量。

s1 := make([]int, 5)          // 长度为5,容量为5,元素初始化为0
s2 := make([]int, 3, 10)      // 长度为3,容量为10

3. 字面量初始化

可以在声明时通过字面量赋值初始化切片。

s3 := []string{"Go", "Python", "Java"}

4. 从数组或其他切片创建切片

arr := [5]int{1, 2, 3, 4, 5}
s4 := arr[1:4] // 包含索引1、2、3,即 [2, 3, 4]

5. 使用append函数扩展切片

切片的长度可以通过append函数动态增长。

s := []int{1, 2, 3}
s = append(s, 4, 5) // s现在为 [1, 2, 3, 4, 5]

完整示例:

package main

import "fmt"

func main() {
    // 使用make创建切片
    s1 := make([]int, 5)
    fmt.Println("s1:", s1) // 输出: [0 0 0 0 0]

    s2 := make([]int, 3, 10)
    fmt.Println("s2:", s2) // 输出: [0 0 0]

    // 字面量初始化
    s3 := []string{"Go", "Python", "Java"}
    fmt.Println("s3:", s3) // 输出: [Go Python Java]

    // 从数组创建切片
    arr := [5]int{1, 2, 3, 4, 5}
    s4 := arr[1:4]
    fmt.Println("s4:", s4) // 输出: [2 3 4]

    // 使用append扩展切片
    s4 = append(s4, 6, 7)
    fmt.Println("s4 after append:", s4) // 输出: [2 3 4 6 7]
}

输出:

s1: [0 0 0 0 0]
s2: [0 0 0]
s3: [Go Python Java]
s4: [2 3 4]
s4 after append: [2 3 4 6 7]
切片的操作

1. 添加元素

使用append函数向切片添加元素,可以添加单个或多个元素。

package main

import "fmt"

func main() {
    s := []int{1, 2, 3}
    s = append(s, 4)
    fmt.Println("添加一个元素:", s) // 输出: [1 2 3 4]

    s = append(s, 5, 6)
    fmt.Println("添加多个元素:", s) // 输出: [1 2 3 4 5 6]
}

2. 删除元素

Go语言没有内置的删除函数,但可以通过切片操作实现。

示例:删除索引为2的元素

package main

import "fmt"

func main() {
    s := []int{1, 2, 3, 4, 5}
    index := 2 // 删除元素3

    s = append(s[:index], s[index+1:]...)
    fmt.Println("删除元素后的切片:", s) // 输出: [1 2 4 5]
}

3. 修改元素

直接通过索引修改切片中的元素。

package main

import "fmt"

func main() {
    s := []string{"apple", "banana", "cherry"}
    s[1] = "blueberry"
    fmt.Println("修改后的切片:", s) // 输出: [apple blueberry cherry]
}

4. 切片截取

通过切片操作可以创建子切片,指定起始和结束索引。

package main

import "fmt"

func main() {
    s := []int{10, 20, 30, 40, 50}

    sub1 := s[1:4]
    fmt.Println("sub1:", sub1) // 输出: [20 30 40]

    sub2 := s[:3]
    fmt.Println("sub2:", sub2) // 输出: [10 20 30]

    sub3 := s[2:]
    fmt.Println("sub3:", sub3) // 输出: [30 40 50]
}

5. 复制切片

使用copy函数复制切片内容。

package main

import "fmt"

func main() {
    src := []int{1, 2, 3, 4, 5}
    dst := make([]int, len(src))

    copy(dst, src)
    fmt.Println("源切片:", src)
    fmt.Println("目标切片:", dst)
}

输出:

源切片: [1 2 3 4 5]
目标切片: [1 2 3 4 5]

6. 切片的容量

切片的容量是从切片的起始位置到底层数组末尾的元素数量。使用cap函数可以获取切片的容量。

package main

import "fmt"

func main() {
    s := make([]int, 3, 5)
    fmt.Println("切片:", s)              // 输出: [0 0 0]
    fmt.Println("长度:", len(s))        // 输出: 3
    fmt.Println("容量:", cap(s))        // 输出: 5

    s = append(s, 1, 2)
    fmt.Println("切片 after append:", s) // 输出: [0 0 0 1 2]
    fmt.Println("长度:", len(s))          // 输出: 5
    fmt.Println("容量:", cap(s))          // 输出: 5

    // 再次添加元素,容量会自动增长
    s = append(s, 3)
    fmt.Println("切片 after second append:", s) // 输出: [0 0 0 1 2 3]
    fmt.Println("长度:", len(s))                // 输出: 6
    fmt.Println("容量:", cap(s))                // 输出: 10 (通常会翻倍)
}

输出:

切片: [0 0 0]
长度: 3
容量: 5
切片 after append: [0 0 0 1 2]
长度: 5
容量: 5
切片 after second append: [0 0 0 1 2 3]
长度: 6
容量: 10
切片的底层原理

切片在Go语言中是一个引用类型,包含三个部分:

  1. 指针:指向底层数组的第一个元素。
  2. 长度(len):切片中的元素数量。
  3. 容量(cap):从切片的起始位置到底层数组末尾的元素数量。

示例:

package main

import "fmt"

func main() {
    arr := [5]int{1, 2, 3, 4, 5}
    s := arr[1:4]

    fmt.Printf("数组: %v\n", arr)
    fmt.Printf("切片: %v, len=%d, cap=%d\n", s, len(s), cap(s)) // 输出: [2 3 4], len=3, cap=4

    // 修改切片中的元素
    s[0] = 20
    fmt.Println("修改后的数组:", arr) // 输出: [1 20 3 4 5]
}

输出:

数组: [1 2 3 4 5]
切片: [2 3 4], len=3, cap=4
修改后的数组: [1 20 3 4 5]

解释:

  • 切片s指向数组arr的索引1到3。
  • 修改切片中的元素也会影响底层数组。

容量的影响:

  • 当切片的容量足够时,使用append不会重新分配底层数组。
  • 当容量不足时,append会分配一个新的底层数组,将原有数据复制过来。

示例:

package main

import "fmt"

func main() {
    arr := [3]int{1, 2, 3}
    s := arr[:]

    fmt.Printf("切片: %v, len=%d, cap=%d\n", s, len(s), cap(s)) // 输出: [1 2 3], len=3, cap=3

    // 使用append添加元素,容量不足,会创建新数组
    s = append(s, 4)
    fmt.Printf("切片 after append: %v, len=%d, cap=%d\n", s, len(s), cap(s)) // 输出: [1 2 3 4], len=4, cap=6

    // 修改新切片,不影响原数组
    s[0] = 10
    fmt.Println("切片 after modification:", s) // 输出: [10 2 3 4]
    fmt.Println("原数组:", arr)                // 输出: [1 2 3]
}

输出:

切片: [1 2 3], len=3, cap=3
切片 after append: [1 2 3 4], len=4, cap=6
切片 after modification: [10 2 3 4]
原数组: [1 2 3]

解释:

  • 初始切片s的容量为3。
  • append操作导致切片容量增长,并分配了新的底层数组。
  • 修改新切片不影响原数组。
注意事项
  • 切片与数组的关系:切片是对数组的引用,修改切片会影响底层数组,反之亦然。
  • 内存管理:切片本身不存储数据,数据存储在底层数组中。切片可以通过多个切片引用同一个底层数组,可能导致数据共享和竞态条件。
  • 切片的零值var s []int声明的切片是nil,长度和容量均为0。可以通过appendmake初始化切片。

6.3 Map

Map是键值对的无序集合,键和值可以是不同的类型。Map在Go中作为内置数据类型提供,类似于Python的字典或Java的HashMap。它在快速查找、插入和删除数据方面表现出色。

Map 的声明与使用

1. 声明Map

使用var关键字声明Map时,需要指定键和值的类型。

var capitals map[string]string

解释:

  • capitals是一个键类型为string,值类型为string的Map。
  • 初始值为nil,需要使用make函数初始化。

2. 使用make初始化Map

capitals = make(map[string]string)

3. 声明并初始化Map

可以在声明时通过字面量赋值初始化Map。

capitals := map[string]string{
    "中国": "北京",
    "美国": "华盛顿",
    "日本": "东京",
}

4. 添加和访问元素

通过键访问或添加元素。

capitals["德国"] = "柏林" // 添加元素
capital := capitals["美国"] // 访问元素
fmt.Println("美国的首都是:", capital) // 输出: 美国的首都是: 华盛顿

5. 完整示例

package main

import "fmt"

func main() {
    // 声明并初始化Map
    capitals := map[string]string{
        "中国": "北京",
        "美国": "华盛顿",
        "日本": "东京",
    }
    fmt.Println("原始Map:", capitals)

    // 添加元素
    capitals["德国"] = "柏林"
    fmt.Println("添加德国后的Map:", capitals)

    // 访问元素
    capital := capitals["美国"]
    fmt.Println("美国的首都是:", capital)

    // 修改元素
    capitals["日本"] = "大阪"
    fmt.Println("修改日本后的Map:", capitals)

    // 删除元素
    delete(capitals, "德国")
    fmt.Println("删除德国后的Map:", capitals)
}

输出:

原始Map: map[中国:北京 美国:华盛顿 日本:东京]
添加德国后的Map: map[中国:北京 美国:华盛顿 德国:柏林 日本:东京]
美国的首都是: 华盛顿
修改日本后的Map: map[中国:北京 美国:华盛顿 德国:柏林 日本:大阪]
删除德国后的Map: map[中国:北京 美国:华盛顿 日本:大阪]
Map 的遍历与修改

1. 遍历Map

使用for循环结合range关键字遍历Map。

package main

import "fmt"

func main() {
    capitals := map[string]string{
        "中国": "北京",
        "美国": "华盛顿",
        "日本": "东京",
    }

    for country, capital := range capitals {
        fmt.Printf("%s 的首都是 %s\n", country, capital)
    }
}

输出示例:

中国 的首都是 北京
美国 的首都是 华盛顿
日本 的首都是 东京

2. 仅遍历键或值

如果只需要键或值,可以使用_忽略不需要的部分。

仅遍历键:

for country := range capitals {
    fmt.Println("国家:", country)
}

仅遍历值:

for _, capital := range capitals {
    fmt.Println("首都:", capital)
}

3. 修改Map元素

在遍历过程中可以直接修改Map的元素。

package main

import "fmt"

func main() {
    capitals := map[string]string{
        "中国": "北京",
        "美国": "华盛顿",
        "日本": "东京",
    }

    // 修改所有首都名称
    for country := range capitals {
        capitals[country] = "首都-" + capitals[country]
    }

    fmt.Println("修改后的Map:", capitals)
}

输出:

修改后的Map: map[中国:首都-北京 美国:首都-华盛顿 日本:首都-东京]

4. 检查键是否存在

在访问Map的元素时,可以同时检查键是否存在。

package main

import "fmt"

func main() {
    capitals := map[string]string{
        "中国": "北京",
        "美国": "华盛顿",
    }

    capital, exists := capitals["日本"]
    if exists {
        fmt.Println("日本的首都是:", capital)
    } else {
        fmt.Println("日本的首都不存在")
    }
}

输出:

日本的首都不存在

5. 使用delete函数删除元素

package main

import "fmt"

func main() {
    capitals := map[string]string{
        "中国": "北京",
        "美国": "华盛顿",
        "日本": "东京",
    }

    delete(capitals, "美国")
    fmt.Println("删除美国后的Map:", capitals)
}

输出:

删除美国后的Map: map[中国:北京 日本:东京]
注意事项
  • Map的零值:未初始化的Map为nil,不能进行读写操作。需要使用make或字面量初始化Map。

    var m map[string]int
    // m["key"] = 1 // 运行时错误: assignment to entry in nil map
    
    m = make(map[string]int)
    m["key"] = 1 // 正确
    
  • Map的无序性:Map中的元素是无序的,遍历时元素的顺序是不确定的。如果需要有序的数据结构,建议使用切片或其他结构。

    示例:

    package main
    
    import "fmt"
    
    func main() {
        m := map[string]int{
            "apple":  5,
            "banana": 3,
            "cherry": 7,
        }
    
        for k, v := range m {
            fmt.Printf("%s: %d\n", k, v)
        }
        // 输出顺序不确定
    }
    
  • Map的键类型:Map的键必须是可比较的类型,如布尔型、数字、字符串、指针、接口和结构体(前提是结构体的所有字段都是可比较的)。切片、Map和函数类型不能作为键。

    // 合法键类型
    m1 := map[string]int{}
    m2 := map[int]bool{}
    m3 := map[struct{ a int; b string }]float64{}
    
    // 非法键类型
    // m4 := map[[]int]string{}      // 编译错误: invalid map key type []int
    // m5 := map[map[string]int]int{} // 编译错误: invalid map key type map[string]int
    // m6 := map[func(){}]bool{}     // 编译错误: invalid map key type func()
    

6.4 结构体

结构体是由多个字段组成的复合数据类型,可以包含不同类型的数据。结构体在Go语言中用于创建自定义的数据类型,方便组织和管理复杂的数据。

定义结构体

使用type关键字定义结构体。

基本语法:

type StructName struct {
    Field1 Type1
    Field2 Type2
    // ...
}

示例:

type Person struct {
    Name string
    Age  int
}

嵌入结构体

结构体可以嵌入其他结构体,实现类似继承的功能。

type Address struct {
    City    string
    ZipCode string
}

type Employee struct {
    Person
    Address
    Position string
}
结构体实例化

1. 使用字面量

p1 := Person{Name: "Alice", Age: 30}

2. 不指定字段名

p2 := Person{"Bob", 25}

3. 使用new关键字

new函数返回指向新分配的零值的指针。

p3 := new(Person)
p3.Name = "Charlie"
p3.Age = 28

4. 部分初始化

未初始化的字段会使用类型的零值。

p4 := Person{Name: "Diana"}
fmt.Println(p4.Age) // 输出: 0

完整示例:

package main

import "fmt"

// 定义结构体
type Person struct {
    Name string
    Age  int
}

func main() {
    // 使用字面量初始化
    p1 := Person{Name: "Alice", Age: 30}
    fmt.Println("p1:", p1)

    // 不指定字段名
    p2 := Person{"Bob", 25}
    fmt.Println("p2:", p2)

    // 使用new关键字
    p3 := new(Person)
    p3.Name = "Charlie"
    p3.Age = 28
    fmt.Println("p3:", *p3)

    // 部分初始化
    p4 := Person{Name: "Diana"}
    fmt.Println("p4:", p4)
}

输出:

p1: {Alice 30}
p2: {Bob 25}
p3: {Charlie 28}
p4: {Diana 0}
嵌套结构体

结构体可以嵌入其他结构体,实现数据的层次化管理。

package main

import "fmt"

// 定义Address结构体
type Address struct {
    City    string
    ZipCode string
}

// 定义Person结构体
type Person struct {
    Name    string
    Age     int
    Address Address
}

func main() {
    p := Person{
        Name: "Eve",
        Age:  35,
        Address: Address{
            City:    "New York",
            ZipCode: "10001",
        },
    }

    fmt.Println("Person:", p)
    fmt.Println("City:", p.Address.City)
}

输出:

Person: {Eve 35 {New York 10001}}
City: New York

匿名嵌入结构体

通过匿名字段,可以直接访问嵌套结构体的字段,类似于继承。

package main

import "fmt"

// 定义Address结构体
type Address struct {
    City    string
    ZipCode string
}

// 定义Person结构体,匿名嵌入Address
type Person struct {
    Name string
    Age  int
    Address
}

func main() {
    p := Person{
        Name: "Frank",
        Age:  40,
        Address: Address{
            City:    "Los Angeles",
            ZipCode: "90001",
        },
    }

    fmt.Println("Person:", p)
    fmt.Println("City:", p.City) // 直接访问嵌套结构体的字段
}

输出:

Person: {Frank 40 {Los Angeles 90001}}
City: Los Angeles
方法与结构体

Go语言支持为结构体类型定义方法,使得结构体更具行为性。

1. 定义方法

方法是在特定类型上定义的函数。通过方法,可以操作结构体的字段。

基本语法:

func (receiver StructType) MethodName(params) returnTypes {
    // 方法体
}

示例:

package main

import "fmt"

// 定义结构体
type Rectangle struct {
    Width, Height float64
}

// 定义方法计算面积
func (r Rectangle) Area() float64 {
    return r.Width * r.Height
}

// 定义方法计算周长
func (r Rectangle) Perimeter() float64 {
    return 2*(r.Width + r.Height)
}

func main() {
    rect := Rectangle{Width: 10, Height: 5}
    fmt.Println("面积:", rect.Area())         // 输出: 面积: 50
    fmt.Println("周长:", rect.Perimeter())   // 输出: 周长: 30
}

2. 方法的接收者

接收者可以是值类型或指针类型。使用指针接收者可以修改结构体的字段,避免复制整个结构体。

示例:

package main

import "fmt"

// 定义结构体
type Counter struct {
    count int
}

// 值接收者方法
func (c Counter) Increment() {
    c.count++
    fmt.Println("Inside Increment (value receiver):", c.count)
}

// 指针接收者方法
func (c *Counter) IncrementPointer() {
    c.count++
    fmt.Println("Inside IncrementPointer (pointer receiver):", c.count)
}

func main() {
    c := Counter{count: 10}

    c.Increment() // 修改的是副本
    fmt.Println("After Increment:", c.count) // 输出: 10

    c.IncrementPointer() // 修改的是原始值
    fmt.Println("After IncrementPointer:", c.count) // 输出: 11

    // 使用指针变量
    cp := &c
    cp.IncrementPointer()
    fmt.Println("After cp.IncrementPointer:", c.count) // 输出: 12
}

输出:

Inside Increment (value receiver): 11
After Increment: 10
Inside IncrementPointer (pointer receiver): 11
After IncrementPointer: 11
Inside IncrementPointer (pointer receiver): 12
After cp.IncrementPointer: 12

3. 方法的作用

方法可以提供结构体的行为和操作,增强代码的可读性和可维护性。例如,可以为结构体定义打印、验证、计算等功能。

示例:验证结构体字段

package main

import (
    "fmt"
    "errors"
)

// 定义结构体
type User struct {
    Username string
    Email    string
    Age      int
}

// 定义方法验证User
func (u *User) Validate() error {
    if u.Username == "" {
        return errors.New("用户名不能为空")
    }
    if u.Email == "" {
        return errors.New("邮箱不能为空")
    }
    if u.Age < 0 || u.Age > 150 {
        return errors.New("年龄不合法")
    }
    return nil
}

func main() {
    user := User{
        Username: "john_doe",
        Email:    "john@example.com",
        Age:      28,
    }

    if err := user.Validate(); err != nil {
        fmt.Println("验证失败:", err)
    } else {
        fmt.Println("用户信息合法")
    }

    // 测试不合法的用户
    invalidUser := User{
        Username: "",
        Email:    "invalid@example.com",
        Age:      200,
    }

    if err := invalidUser.Validate(); err != nil {
        fmt.Println("验证失败:", err) // 输出: 验证失败: 用户名不能为空
    } else {
        fmt.Println("用户信息合法")
    }
}

输出:

用户信息合法
验证失败: 用户名不能为空
注意事项
  • 接收者的选择:根据方法是否需要修改结构体的字段,选择值接收者或指针接收者。一般情况下,使用指针接收者可以避免复制结构体,提升性能,且可以修改结构体的字段。

    // 修改结构体字段
    func (p *Person) SetName(name string) {
        p.Name = name
    }
    
  • 方法的命名:方法名应简洁明了,能够清晰描述方法的功能。例如,CalculateAreaPrintDetails等。

  • 方法与函数的区别:方法是与特定类型相关联的函数,而函数是独立的。合理使用方法可以提升代码的可读性和组织性。

6.5 指针与结构体

指针是存储变量内存地址的变量。在Go语言中,指针与结构体结合使用,可以提高程序的性能,避免大量数据的复制,同时实现对结构体的修改和共享。

指针基础

1. 声明指针

使用*符号声明指针类型。

var p *int

解释:

  • p是一个指向int类型的指针,初始值为nil

2. 获取变量的地址

使用&符号获取变量的内存地址。

a := 10
p := &a
fmt.Println("a的地址:", p) // 输出: a的地址: 0xc0000140b0

3. 解引用指针

使用*符号访问指针指向的值。

fmt.Println("p指向的值:", *p) // 输出: p指向的值: 10

4. 修改指针指向的值

通过指针修改变量的值。

*p = 20
fmt.Println("修改后的a:", a) // 输出: 修改后的a: 20

完整示例:

package main

import "fmt"

func main() {
    var a int = 10
    var p *int = &a

    fmt.Println("变量a的值:", a)         // 输出: 10
    fmt.Println("指针p的地址:", p)       // 输出: a的地址
    fmt.Println("指针p指向的值:", *p)     // 输出: 10

    // 修改指针指向的值
    *p = 30
    fmt.Println("修改后的a:", a)         // 输出: 30
}

输出:

变量a的值: 10
指针p的地址: 0xc0000140b0
指针p指向的值: 10
修改后的a: 30
指针与结构体

将指针与结构体结合使用,可以避免复制整个结构体,尤其是当结构体较大时,提高程序的性能。此外,通过指针,可以在函数中修改结构体的字段。

1. 定义结构体并使用指针

package main

import "fmt"

// 定义结构体
type Person struct {
    Name string
    Age  int
}

func main() {
    p := Person{Name: "Alice", Age: 25}
    fmt.Println("原始结构体:", p) // 输出: {Alice 25}

    // 获取结构体的指针
    ptr := &p

    // 修改指针指向的结构体字段
    ptr.Age = 26
    fmt.Println("修改后的结构体:", p) // 输出: {Alice 26}
}

2. 结构体指针作为函数参数

通过将结构体指针作为函数参数,可以在函数中修改结构体的字段,而无需返回修改后的结构体。

package main

import "fmt"

// 定义结构体
type Rectangle struct {
    Width, Height float64
}

// 定义函数,接受结构体指针并修改字段
func Resize(r *Rectangle, width, height float64) {
    r.Width = width
    r.Height = height
}

func main() {
    rect := Rectangle{Width: 10, Height: 5}
    fmt.Println("原始矩形:", rect) // 输出: {10 5}

    Resize(&rect, 20, 10)
    fmt.Println("修改后的矩形:", rect) // 输出: {20 10}
}

3. 指针与方法接收者

前面章节中提到方法接收者可以是指针类型,这样可以在方法中修改结构体的字段。

package main

import "fmt"

// 定义结构体
type Counter struct {
    count int
}

// 定义指针接收者方法
func (c *Counter) Increment() {
    c.count++
}

func main() {
    c := Counter{count: 0}
    fmt.Println("初始计数:", c.count) // 输出: 0

    c.Increment()
    fmt.Println("计数 after Increment:", c.count) // 输出: 1

    // 使用指针变量
    cp := &c
    cp.Increment()
    fmt.Println("计数 after cp.Increment:", c.count) // 输出: 2
}

输出:

初始计数: 0
计数 after Increment: 1
计数 after cp.Increment: 2
指针的高级用法

1. 指针与切片

切片本身是一个引用类型,包含指向底层数组的指针。可以通过指针修改切片元素。

package main

import "fmt"

func main() {
    s := []int{1, 2, 3}
    ptr := &s

    // 修改切片元素
    (*ptr)[1] = 20
    fmt.Println("修改后的切片:", s) // 输出: [1 20 3]
}

2. 指针与Map

Map是引用类型,使用指针传递Map不会带来额外的性能开销。通常不需要使用指针传递Map,但在某些情况下可以提高灵活性。

package main

import "fmt"

func main() {
    capitals := make(map[string]string)
    capitals["中国"] = "北京"
    capitals["美国"] = "华盛顿"

    modifyMap(&capitals)
    fmt.Println("修改后的Map:", capitals) // 输出: map[中国:北京 美国:纽约]
}

func modifyMap(m *map[string]string) {
    (*m)["美国"] = "纽约"
}

3. 指针数组

数组中可以存储指针类型的元素,适用于需要引用和共享数据的场景。

package main

import "fmt"

func main() {
    a, b, c := 1, 2, 3
    ptrArr := []*int{&a, &b, &c}

    for i, ptr := range ptrArr {
        fmt.Printf("ptrArr[%d] 指向的值: %d\n", i, *ptr)
    }

    // 修改通过指针数组修改原始变量
    *ptrArr[0] = 10
    fmt.Println("修改后的a:", a) // 输出: 10
}

输出:

ptrArr[0] 指向的值: 1
ptrArr[1] 指向的值: 2
ptrArr[2] 指向的值: 3
修改后的a: 10
注意事项
  • 指针的零值:未初始化的指针为nil。在使用指针前,确保其已被正确初始化,避免运行时错误。

    var p *int
    // fmt.Println(*p) // 运行时错误: invalid memory address or nil pointer dereference
    
  • 避免悬挂指针:确保指针指向的变量在指针使用期间保持有效,避免指针指向已经释放或超出作用域的变量。

    func getPointer() *int {
        x := 10
        return &x
    }
    
    func main() {
        p := getPointer()
        fmt.Println(*p) // 不安全:x已经超出作用域,可能导致未定义行为
    }
    
  • 使用指针优化性能:对于大型结构体,使用指针传递可以避免复制整个结构体,提高性能。

    type LargeStruct struct {
        Data [1000]int
    }
    
    func process(ls LargeStruct) { // 复制整个结构体
        // ...
    }
    
    func processPointer(ls *LargeStruct) { // 传递指针
        // ...
    }
    
  • nil指针检查:在使用指针前,最好检查指针是否为nil,以避免运行时错误。

    if p != nil {
        fmt.Println(*p)
    } else {
        fmt.Println("指针为nil")
    }
    

6.6 组合与接口(拓展内容)

虽然用户没有列出组合与接口,在数据结构与集合章节中,了解结构体的组合以及接口的使用也是非常重要的。因此,这里提供对组合和接口的简要介绍。

组合(Composition)

组合是通过嵌入一个结构体到另一个结构体中,实现代码复用和功能扩展的一种方式。通过组合,可以创建复杂的数据结构,同时保持代码的简洁和模块化。

示例:

package main

import "fmt"

// 定义基本结构体
type Address struct {
    City    string
    ZipCode string
}

// 定义复合结构体,通过组合Address
type Person struct {
    Name    string
    Age     int
    Address // 组合
}

func main() {
    p := Person{
        Name: "Grace",
        Age:  28,
        Address: Address{
            City:    "San Francisco",
            ZipCode: "94105",
        },
    }

    fmt.Printf("Person: %+v\n", p)
    fmt.Println("City:", p.City) // 直接访问组合结构体的字段
}

输出:

Person: {Name:Grace Age:28 Address:{City:San Francisco ZipCode:94105}}
City: San Francisco

优势:

  • 代码复用:通过组合,可以复用已有的结构体,减少重复代码。
  • 灵活性:组合比继承更灵活,避免了继承带来的复杂性。
接口(Interface)

接口定义了一组方法签名,任何实现了这些方法的类型都满足该接口。接口提供了多态性,使得代码更加灵活和可扩展。

示例:

package main

import "fmt"

// 定义接口
type Greeter interface {
    Greet(name string) string
}

// 定义实现接口的结构体
type EnglishGreeter struct{}

func (eg EnglishGreeter) Greet(name string) string {
    return "Hello, " + name + "!"
}

type ChineseGreeter struct{}

func (cg ChineseGreeter) Greet(name string) string {
    return "你好," + name + "!"
}

func main() {
    var g Greeter

    g = EnglishGreeter{}
    fmt.Println(g.Greet("Alice")) // 输出: Hello, Alice!

    g = ChineseGreeter{}
    fmt.Println(g.Greet("Bob"))   // 输出: 你好,Bob!
}

输出:

Hello, Alice!
你好,Bob!

解释:

  • Greeter接口定义了一个Greet方法。
  • EnglishGreeterChineseGreeter结构体实现了Greet方法,满足Greeter接口。
  • 通过接口类型变量g,可以调用不同实现的Greet方法,实现多态性。

接口的优势:

  • 解耦合:通过接口,可以将代码模块之间的依赖解耦,提高代码的灵活性和可维护性。
  • 多态性:同一接口可以由不同类型实现,允许不同的对象以统一的方式被处理。
  • 可扩展性:无需修改现有代码,只需实现新的接口即可扩展功能。
注意事项
  • 接口隐式实现:在Go语言中,类型只需实现接口的方法,不需要显式声明实现关系。这种隐式实现提高了代码的灵活性和简洁性。

    type Reader interface {
        Read(p []byte) (n int, err error)
    }
    
    type MyReader struct{}
    
    func (r MyReader) Read(p []byte) (n int, err error) {
        // 实现Read方法
        return 0, nil
    }
    
    func main() {
        var r Reader
        r = MyReader{}
    }
    
  • 空接口(interface{}:空接口可以表示任何类型,是实现通用数据结构和函数的重要工具。

    func printAnything(a interface{}) {
        fmt.Println(a)
    }
    
    func main() {
        printAnything(100)
        printAnything("Hello")
        printAnything(true)
    }
    

    输出:

    100
    Hello
    true
    
  • 类型断言和类型切换:在使用接口时,可能需要进行类型断言或类型切换,以访问具体类型的方法或字段。

    类型断言示例:

    func main() {
        var i interface{} = "Go Language"
    
        s, ok := i.(string)
        if ok {
            fmt.Println("字符串长度:", len(s))
        } else {
            fmt.Println("不是字符串类型")
        }
    }
    

    类型切换示例:

    func main() {
        var i interface{} = 3.14
    
        switch v := i.(type) {
        case int:
            fmt.Println("整数:", v)
        case float64:
            fmt.Println("浮点数:", v)
        case string:
            fmt.Println("字符串:", v)
        default:
            fmt.Println("未知类型")
        }
    }
    

网站公告

今日签到

点亮在社区的每一天
去签到