Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go可变参数函数

Go可变参数函数的实现

作者:kobe_OKOK_

可变参数函数是指函数参数的某个参数可有可无,即这个参数的个数可以为0会多个,可变参数函数参数在日常编程中大量使用,本文主要介绍了Go可变参数函数的实现,感兴趣的可以了解一下

1 定义

可变参数函数是一种参数个数可变的函数。

2 语法

语法

//关键字 函数名(参数1, elems为T类型的可变参数) 返回值类型
func name(params type, elems ...T) returntype{
	// 函数体
}

slice的append就是一个可变参数的内部函数

func append(slice []Type, elems ...Type) []Type

可变参数示例1

package main

import "fmt"

func find(target int, group ...int) int {
	ishit := false
	index := 0
	for i, v := range group {
		if target == v {
			fmt.Printf("找到元素值:%d,在索引位置:%d\n", v, i)
			ishit = true
			index = i
		}
	}
	if !ishit {
		return -1
	} else {
		return index
	}

}

func main() {
	find(7, 1, 2, 3, 4, 5)
}

可变参数示例2

package main

import "fmt"

func add(base int, group ...int) {
	total := 0
	for _, v := range group {
		total += v
	}
	total += base
	fmt.Println("total = ", total)
}

func main() {
	add(100, 1, 2, 3, 4, 5)
}

可变参数函数的工作原理是把可变参数转换为一个新的切片,然后被传入函数中。

3 给可变函数参数传入切片

package main

import "fmt"

func find(a int, elems ...int) {
	for i, v := range elems {
		if v == a {
			fmt.Printf("found value %d, at index %d", v, i)
			break
		}
	}
}

func main() {
	a := []int{1, 2, 3, 4, 5}
	find(1, a)  // cannot use a (type []int) as type int in argument to find


}

将切片直接传递给可变参数明显是不可以的,但是可以通过一个语法糖解决这个问题。

将切片传入函数的可变参数位置报错的原因分析

多变参数,将接收到的参数做为一个切片,如果直接传递一个切片会出现下面这种情况:

// 1 函数原型 
func find(a int, elems ...int) 

// 2 函数参数 为切片a 传递到find 等同于 []int{a} 
// 因为切片要求的元素类型为int,传入的确是一个[]int,所以报错
cannot use a (type []int) as type int in argument to find
// 3 go给出的报错信息很友好

如何能够将切片直接传递给可变参数函数?
有一个可以直接将切片传入可变参数函数的语法糖,你可以在在切片后加上 … 后缀。如果这样做,切片将直接传入函数,不再创建新的切片

修改上面的函数

package main

import "fmt"

func find(a int, elems ...int) {
	for i, v := range elems {
		if v == a {
			fmt.Printf("found value %d, at index %d", v, i)
			break
		}
	}
}

func main() {
	a := []int{1, 2, 3, 4, 5}
	find(1, a...) // cannot use a (type []int) as type int in argument to find

}

4 修改可变参数函数中的切片

如果使用切片…,那么会将切片直接传递给函数,不会再创建新的切片,这也意味着,如果再可变参数函数内部修改了切片,会影响到创建切片的位置,这可能比较抽象,直接看一个例子就好了。

示例一

package main

import "fmt"

func changeTest(s ...string) {
	s[0] = "上树捉鸟"
	s[1] = "下河洗澡"
}

func main() {
	s := []string{"好好学习", "天天向上"}
	fmt.Println("传递前: ", s)
	changeTest(s...) // 将切片传递过去,changeTest不会再创建切片
	fmt.Println("传递后: ", s)
}

示例一执行结果:

传递前:  [好好学习 天天向上]
传递后:  [上树捉鸟 下河洗澡]

示例2

package main

import "fmt"

func changeTest1(s ...string) {
	s[0] = "上树捉鸟"
	s[1] = "下河洗澡"
	s = append(s, "窈窕淑女", "君子好逑")
	fmt.Println("in changeTest: ", s)
}

func main() {
	s := []string{"好好学习", "天天向上"}
	fmt.Println("传递前: ", s)
	changeTest1(s...) // 将切片传递过去,changeTest不会再创建切片
	fmt.Println("传递后: ", s)
}

示例2执行结果:

传递前:  [好好学习 天天向上]
in changeTest:  [上树捉鸟 下河洗澡 窈窕淑女 君子好逑]
传递后:  [上树捉鸟 下河洗澡]

到此这篇关于Go可变参数函数的实现的文章就介绍到这了,更多相关Go可变参数函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文