Go语言圣经-函数递归习题详解编程语言

// Findlinks1 prints the links in an HTML document read from standard input. 
package main 
 
import ( 
        "fmt" 
        "os" 
 
        "golang.org/x/net/html" 
) 
 
func main() { 
        doc, err := html.Parse(os.Stdin) 
        if err != nil { 
                fmt.Fprintf(os.Stderr, "findlinks1: %v/n", err) 
                os.Exit(1) 
        } 
        for _, link := range visit(nil, doc) { 
                fmt.Println(link) 
        } 
 
        var res = make(map[string]int) 
        res = count(res, doc) 
        for k, v := range res { 
                fmt.Printf("%s==>%d /n", k, v) 
        } 
        //fmt.Println(res) 
        for _, text := range visit3(nil, doc) { 
                fmt.Println(text) 
        } 
 
        for _, link := range visit4(nil, doc) { 
                fmt.Println(link) 
        } 
 
} 
 
// visit appends to links each link found in n and returns the result. 
func visit(links []string, n *html.Node) []string { 
        if n.Type == html.ElementNode && n.Data == "a" { 
                for _, a := range n.Attr { 
                        if a.Key == "href" { 
                                links = append(links, a.Val) 
                        } 
                } 
        } 
        /* 
           练习 5.1： 修改findlinks代码中遍历n.FirstChild链表的部分，将循环调用visit，改成递归调用。 
           实在是不知道为啥不对，我选择放弃 
                if n.FirstChild!=nil{ 
                        links=visit(links,n.FirstChild) 
                }else if n.NextSibling!=nil{ 
                        //n=n.NextSibling 
                        links=visit(links,n.NextSibling) 
                } 
        */ 
        for c := n.FirstChild; c != nil; c = c.NextSibling { 
                links = visit(links, c) 
        } 
        return links 
} 
 
/* 
练习 5.2： 编写函数，记录在HTML树中出现的同名元素的次数。 
*/ 
func count(res map[string]int, n *html.Node) map[string]int { 
        if n.Type == html.ElementNode { 
                res[n.Data]++ 
        } 
        for c := n.FirstChild; c != nil; c = c.NextSibling { 
                res = count(res, c) 
        } 
        return res 
} 
 
/* 
练习 5.3： 编写函数输出所有text结点的内容。注意不要访问<script>和<style>元素,因为这些元素对浏览者是不可见的。 
*/ 
func visit3(texts []string, n *html.Node) []string { 
        if n.Type == html.TextNode { 
                texts = append(texts, n.Data) 
        } 
        for c := n.FirstChild; c != nil; c = c.NextSibling { 
                if c.Data == "script" || c.Data == "style" { 
                        continue 
                } 
 
                texts = visit3(texts, c) 
        } 
        return texts 
} 
 
/* 
练习 5.4： 扩展visit函数，使其能够处理其他类型的结点，如images、scripts和style sheets。 
*/ 
func visit4(links []string, n *html.Node) []string { 
        if n.Type == html.ElementNode && (n.Data == "a" || n.Data == "img" || n.Data == "link" || n.Data == "scripts") { 
                for _, a := range n.Attr { 
                        if a.Key == "href" { 
                                links = append(links, a.Val) 
                        } 
                } 
        } 
        for c := n.FirstChild; c != nil; c = c.NextSibling { 
                links = visit4(links, c) 
        } 
        return links 
}