main.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "io"
  6. "os"
  7. "strings"
  8. "time"
  9. )
  10. func main() {
  11. // rand.Seed(time.Now().UnixNano())
  12. // b := bst.New()
  13. // // for i := 0; i < 100; i++ {
  14. // // b.Add(rand.Intn(1000))
  15. // // }
  16. // b.Add(10)
  17. // b.Add(5)
  18. // b.Add(15)
  19. // b.Add(6)
  20. // b.Add(4)
  21. // b.Add(3)
  22. // b.Add(8)
  23. // b.Add(7)
  24. // b.Add(16)
  25. // b.Add(14)
  26. // b.Add(13)
  27. // b.Add(18)
  28. // b.Add(17)
  29. // b.GetList()
  30. // fmt.Println(b.Contains(8))
  31. // fmt.Println(b.Minimum())
  32. // fmt.Println(b.Maxmum())
  33. // b.Remove(8)
  34. // b.GetList()
  35. // fmt.Println(b.Contains(8))
  36. // filename, _ := filepath.Abs("a.txt")
  37. // fmt.Println(filename)
  38. // words1 := ReadFile(filename)
  39. // fmt.Println(len(words1))
  40. // set1 := bstset.New()
  41. // for _, word := range words1 {
  42. // set1.Add(word)
  43. // }
  44. // fmt.Println(set1.GetSize())
  45. // fmt.Println("a Tale of Two Cities")
  46. // filename, _ = filepath.Abs("b.txt")
  47. // words2 := ReadFile(filename)
  48. // fmt.Println(len(words2))
  49. // set2 := bstset.New()
  50. // for _, word := range words2 {
  51. // set2.Add(word)
  52. // }
  53. // fmt.Println(set2.GetSize())
  54. // maxheaptest := maxheap.New()
  55. // rand.Seed(time.Now().UnixNano())
  56. // for i := 0; i < 1000; i++ {
  57. // maxheaptest.Add(rand.Intn(1000000))
  58. // }
  59. // n := 0
  60. // m := maxheaptest.GetSize()
  61. // for i := 0; i < 1000; i++ {
  62. // fmt.Println(maxheaptest.ExtractMax(), maxheaptest.GetSize())
  63. // n++
  64. // }
  65. // fmt.Println(n, m)
  66. timeTemplate := "2006-01-02"
  67. t, _ := time.ParseInLocation(timeTemplate, "2010-01-01", time.Local)
  68. fmt.Println(int(t.Weekday()), t.AddDate(0, 1, -1))
  69. }
  70. func ReadFile(filename string) []string {
  71. var words []string
  72. file, err := os.Open(filename)
  73. if err != nil {
  74. panic(err)
  75. }
  76. defer file.Close()
  77. reader := bufio.NewReader(file)
  78. for {
  79. line, err := reader.ReadString('\n')
  80. if err != nil || io.EOF == err {
  81. break
  82. }
  83. wordSlice := strings.Fields(line)
  84. for _, word := range wordSlice {
  85. if word = extractStr(strings.ToLower(word)); word != "" {
  86. words = append(words, word)
  87. }
  88. }
  89. }
  90. return words
  91. }
  92. func extractStr(str string) string {
  93. var res []rune
  94. for _, letter := range str {
  95. if letter >= 'a' && letter <= 'z' {
  96. res = append(res, letter)
  97. }
  98. }
  99. return string(res)
  100. }