Add person data statistic

ctl/util.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"sort"
+	"strconv"
+)
+
+func loadSolutionsDir() ([]int, int) {
+	files, err := ioutil.ReadDir("../leetcode/")
+	if err != nil {
+		fmt.Println(err)
+	}
+	solutionIds := []int{}
+	for _, f := range files {
+		if f.Name()[4] == '.' {
+			tmp, err := strconv.Atoi(f.Name()[:4])
+			if err != nil {
+				fmt.Println(err)
+			}
+			solutionIds = append(solutionIds, tmp)
+		}
+	}
+	sort.Ints(solutionIds)
+	fmt.Printf("读取了 %v 道题的题解，当前目录下有 %v 个文件(可能包含 .DS_Store)，目录中有 %v 道题在尝试中\n", len(solutionIds), len(files), len(files)-len(solutionIds))
+	return solutionIds, len(files) - len(solutionIds)
+}
+
+func writeFile(fileName string, content []byte) {
+	file, err := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE, 0777)
+	if err != nil {
+		fmt.Println(err)
+	}
+	defer file.Close()
+
+	_, err = file.Write(content)
+	if err != nil {
+		fmt.Println(err)
+	}
+	fmt.Println("write file successful")
+}
+
+func binarySearch(nums []int, target int) int {
+	low, high := 0, len(nums)-1
+	for low <= high {
+		mid := low + (high-low)>>1
+		if nums[mid] == target {
+			return mid
+		} else if nums[mid] > target {
+			high = mid - 1
+		} else {
+			low = mid + 1
+		}
+	}
+	return -1
+}