在 Go 中表达函数的更好方法( struct 方法)

发布于03月07日

这是一个遍历点数组的函数.点是一个包含int类型元素x和y的 struct .遍历函数是从另一个值为dir的函数调用的.可以避免循环开始时的if判断，因为调用函数后，dir值不会改变.有没有更好的方式来表达围棋中的这种逻辑？struct方法返回一个点.我提供了一个struct方法的示例.

type Point struct {
    x int
    y int
}

func (p Point) next_row() Point {
    nearby := Point{p.x + 1, p.y}
    return nearby
}

func trv_point(p Point, p_list []Point, dir string) int {
    seg_count := 0
    var nearby Point
    for {
        if dir == "up" {
            nearby = p.prev_row()
        }
        if dir == "down" {
            nearby = p.next_row()
        }
        if dir == "left" {
            nearby = p.prev_col()
        }
        if dir == "right" {
            nearby = p.next_col()
        }

        if !check_point_in_slice(p_list, nearby) {
            break
        }
        seg_count++
        p = nearby

    }
    return seg_count
}

推荐答案

因为dir在这里是不变的，这if条语句是冗余的，在这种情况下，你也应该使用if-else条语句，因为只有一条["向上"、"向下"、"向左"、"向右"]可能发生.

type Point struct {
    x int
    y int
}

func (p *Point) next_row() {
    p.x += 1
}

func main() {
    p := Point{0, 0}
    f := p.next_row

    for i := 0; i < 2; i++ {
        fmt.Println(p)
        f()
    }
    fmt.Println(p)
}

输出将是:

{0 0}
{1 0}
{2 0}

这里我们将接收器从(p Point)改为(p *Point)，因为我们想保存x和y的效果.

type Point struct {
    x int
    y int
}

func (p *Point) next_row() {
    p.x += 1
}

func get_move(dir string) func() {
    var f func()
    switch dir {
    case "up":
        f = p.prev_row
    case "down":
        f = p.next_row
    case "left":
        f = p.prev_col
    case "right":
        f = p.next_col
    }
    return f
}

func trv_point(p Point, p_list []Point, dir string) int {
    seg_count := 0
    var nearby Point
    f := get_move(dir)
    for check_point_in_slice(p_list, p) {
        f()
        seg_count++
    }
    return seg_count
}