go学习笔记:计算文件中重复的行,并输出重复次数

方法一:按行读取
package main

import (
"bufio"
"fmt"
"os"
)

func main() {
counts:=make(map[string]int)
files:=os.Args[1:]
if len(files) == 0 {
countLines(os.Stdin,counts)
}else{
for _,arg:=range files{
f,err:=os.Open(arg)
if err != nil {
fmt.Fprintf(os.Stderr,"dup2:%v\n",err)
continue
}
countLines(f,counts)
f.Close()
}
}
for line,n:=range counts{
if n > 1{
fmt.Printf("%d\t%s\n",n,line)
}
}
}

func countLines(f *os.File, counts map[string]int){
input:=bufio.NewScanner(f)
for input.Scan(){
counts[input.Text()]++
}
}
方法二:一次性读入到内存中,再按行处理
package main

import (
"fmt"
"io/ioutil"
"os"
"strings"
)

func main() {
counts:=make(map[string]int)
for ,filename:=range os.Args[1:]{
data,err:=ioutil.ReadFile(filename)
if err!=nil{
fmt.Fprintf(os.Stderr,"dup2:%v\n",err)
continue
}
for
,line:=range strings.Split(string(data),"\n"){
counts[line]++
}
}
for line,n:=range counts{
if n >1{
fmt.Printf("%d\t%s\n",n,line)
}
}
}

猜你喜欢

转载自blog.51cto.com/zhangdl/2537061