使用映射为其设置属性和用户定义的类型

时间:2011-01-16 20:18:17

标签: dictionary go set

我正在尝试使用内置地图类型作为我自己类型的集合(在本例中为Point)。问题是,当我将Point指定给地图,然后稍后创建一个新的但相等的点并将其用作键时,地图的行为就好像该键不在地图中。这不可能吗?

// maptest.go

package main

import "fmt"

func main() {
    set := make(map[*Point]bool)

    printSet(set)
    set[NewPoint(0, 0)] = true
    printSet(set)
    set[NewPoint(0, 2)] = true
    printSet(set)

    _, ok := set[NewPoint(3, 3)] // not in map
    if !ok {
        fmt.Print("correct error code for non existent element\n")
    } else {
        fmt.Print("incorrect error code for non existent element\n")
    }

    c, ok := set[NewPoint(0, 2)] // another one just like it already in map
    if ok {
        fmt.Print("correct error code for existent element\n") // should get this
    } else {
        fmt.Print("incorrect error code for existent element\n") // get this
    }

    fmt.Printf("c: %t\n", c)
}

func printSet(stuff map[*Point]bool) {
    fmt.Print("Set:\n")
    for k, v := range stuff {
        fmt.Printf("%s: %t\n", k, v)
    }
}

type Point struct {
    row int
    col int
}

func NewPoint(r, c int) *Point {
    return &Point{r, c}
}

func (p *Point) String() string {
    return fmt.Sprintf("{%d, %d}", p.row, p.col)
}

func (p *Point) Eq(o *Point) bool {
    return p.row == o.row && p.col == o.col
}

1 个答案:

答案 0 :(得分:2)

package main

import "fmt"

type Point struct {
    row int
    col int
}

func main() {
    p1 := &Point{1, 2}
    p2 := &Point{1, 2}
    fmt.Printf("p1: %p %v p2: %p %v\n", p1, *p1, p2, *p2)

    s := make(map[*Point]bool)
    s[p1] = true
    s[p2] = true
    fmt.Println("s:", s)

    t := make(map[int64]*Point)
t[int64(p1.row)<<32+int64(p1.col)] = p1
t[int64(p2.row)<<32+int64(p2.col)] = p2
    fmt.Println("t:", t)
}

Output:
p1: 0x7fc1def5e040 {1 2} p2: 0x7fc1def5e0f8 {1 2}
s: map[0x7fc1def5e0f8:true 0x7fc1def5e040:true]
t: map[4294967298:0x7fc1def5e0f8]

如果我们使用相同的坐标创建指向两个Points p1p2的指针,则指向不同的地址。

s := make(map[*Point]bool)创建一个映射,其中键是指向分配给Point的内存的指针,值为布尔值。因此,如果我们将元素p1p2分配给地图s,那么我们有两个不同的地图键和两个具有相同坐标的不同地图元素。

t := make(map[int64]*Point)创建一个地图,其中键是Point坐标的合成,而值是指向Point坐标的指针。因此,如果我们将元素p1p2分配给地图t,那么我们有两个相等的地图键和一个带共享坐标的地图元素。