[日常] Go语言圣经-可变参数习题

package main

import (
        "fmt"
        "golang.org/x/net/html"
        "net/http"
)

func main() {
        fmt.Println(max(1, 2, 3, 4, 2)) 
        fmt.Println(max())
        fmt.Println(join("-", "hello", "tsh", "hi"))

        resp, _ := http.Get("http://mail.sina.net")
        doc, _ := html.Parse(resp.Body)
        resp.Body.Close()
        tNodes := ElementsByTagName(doc, "p", "a")
        for _, v := range tNodes {
                ForEachNode(v)
                fmt.Println("------------------------")
        }   
}

/*
练习5.15： 编写类似sum的可变参数函数max和min。考虑不传参时，max和min该如何处理，再编写至少接收1个参数的版本。
*/
func max(vals ...int) (int, error) {
        var m int 

        num := len(vals)
        if num == 0 { 
                return 0, fmt.Errorf("max: %s", "至少传递一个参数")
        }   
        for _, v := range vals {
                if m < v { 
                        m = v 
                }   
        }   
        return m, nil 
}

/*
练习5.16：编写多参数版本的strings.Join。
*/
func join(sep string, strs ...string) (string, error) {
        var res string
        var num = len(strs)
        if num == 0 { 
                return "", fmt.Errorf("join: %s", "至少传递一个参数")
        }   

        for i, v := range strs {
                if i == (num - 1) {
                        res += v
                } else {
                        res += v + sep
                }
        }
        return res, nil
}

/*
练习5.17：编写多参数版本的ElementsByTagName，函数接收一个HTML结点树以及任意数量的标签名，返回与这些标签名匹配的所有元素。下面给出了2个例子：
*/
var nodes []*html.Node

func ElementsByTagName(n *html.Node, names ...string) []*html.Node {
        newNode := n
        for _, name := range names {
                if n.Type == html.ElementNode && n.Data == name {
                        nodes = append(nodes, newNode)
                }
        }
        for c := n.FirstChild; c != nil; c = c.NextSibling {
                //可变参数传参特点
                ElementsByTagName(c, names...)
        }
        return nodes
}
func ForEachNode(n *html.Node) {
        fmt.Println(n.Data)
        for c := n.FirstChild; c != nil; c = c.NextSibling {
                ForEachNode(c)
        }
}