快捷導航

Go可變參數(shù)函數(shù)的實現(xiàn)

更新時間：2023年12月14日 11:36:31 作者：kobe_OKOK_

可變參數(shù)函數(shù)是指函數(shù)參數(shù)的某個參數(shù)可有可無,即這個參數(shù)的個數(shù)可以為0會多個,可變參數(shù)函數(shù)參數(shù)在日常編程中大量使用,本文主要介紹了Go可變參數(shù)函數(shù)的實現(xiàn),感興趣的可以了解一下

1 定義

可變參數(shù)函數(shù)是一種參數(shù)個數(shù)可變的函數(shù)。

2 語法

語法

//關(guān)鍵字 函數(shù)名(參數(shù)1， elems為T類型的可變參數(shù)) 返回值類型
func name(params type, elems ...T) returntype{
	// 函數(shù)體
}

slice的append就是一個可變參數(shù)的內(nèi)部函數(shù)

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

可變參數(shù)示例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)
}

可變參數(shù)示例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)
}

可變參數(shù)函數(shù)的工作原理是把可變參數(shù)轉(zhuǎn)換為一個新的切片，然后被傳入函數(shù)中。

3 給可變函數(shù)參數(shù)傳入切片

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


}

將切片直接傳遞給可變參數(shù)明顯是不可以的，但是可以通過一個語法糖解決這個問題。

將切片傳入函數(shù)的可變參數(shù)位置報錯的原因分析

多變參數(shù)，將接收到的參數(shù)做為一個切片，如果直接傳遞一個切片會出現(xiàn)下面這種情況：

// 1 函數(shù)原型 
func find(a int, elems ...int) 

// 2 函數(shù)參數(shù) 為切片a 傳遞到find 等同于 []int{a} 
// 因為切片要求的元素類型為int，傳入的確是一個[]int,所以報錯
cannot use a (type []int) as type int in argument to find
// 3 go給出的報錯信息很友好

如何能夠?qū)⑶衅苯觽鬟f給可變參數(shù)函數(shù)？
有一個可以直接將切片傳入可變參數(shù)函數(shù)的語法糖，你可以在在切片后加上 … 后綴。如果這樣做，切片將直接傳入函數(shù)，不再創(chuàng)建新的切片

修改上面的函數(shù)

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 修改可變參數(shù)函數(shù)中的切片

如果使用切片…，那么會將切片直接傳遞給函數(shù)，不會再創(chuàng)建新的切片，這也意味著，如果再可變參數(shù)函數(shù)內(nèi)部修改了切片，會影響到創(chuàng)建切片的位置，這可能比較抽象，直接看一個例子就好了。

示例一

package main

import "fmt"

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

func main() {
	s := []string{"好好學習", "天天向上"}
	fmt.Println("傳遞前： ", s)
	changeTest(s...) // 將切片傳遞過去，changeTest不會再創(chuàng)建切片
	fmt.Println("傳遞后： ", s)
}

示例一執(zhí)行結(jié)果：

傳遞前： [好好學習天天向上]
傳遞后： [上樹捉鳥下河洗澡]

示例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不會再創(chuàng)建切片
	fmt.Println("傳遞后： ", s)
}

示例2執(zhí)行結(jié)果：