GO数据结构(一)——稀疏数组

0  0  0  0  0  0  0  0  0  0  0
0  0  1  0  0  0  0  0  0  0  0
0  0  0  2  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0
0  0  0  0  0  0  0  0  0  0  0

func writeSparse()  {
	// 定义一个结构体，用来存放稀疏矩阵的值
	type ValNode struct{
		row int		// 所在行
		col int		// 所在列
		val int		// 值
	}

	// 定义一个二维数组——棋盘
	// 其他都为0
	var chessMap [11][11]int
	chessMap[1][2] = 1
	chessMap[2][3] = 2

	// 定义一个切片
	var sparseArr []ValNode

	// 11行 11列 默认值为0
	valNode := ValNode{
		row: 11,
		col: 11,
		val: 0,
	}
	sparseArr =append(sparseArr,valNode)

	// 遍历棋盘，寻找有棋子的地方
	for i, v:= range chessMap{
		for j, v2:=range v{
			if v2 != 0{
				// 拿到黑子和白子具体的位置
				valNode := ValNode{
					row: i,
					col: j,
					val: v2,
				}
				sparseArr =append(sparseArr,valNode)
			}
		}
	}

	// 存到文件中 便于续盘
	filePath := "./数据结构/sparseArr.txt"
	file, err := os.OpenFile(filePath,os.O_WRONLY|os.O_CREATE,0666)
	// 第一个数字代表u(拥有者)权限，
	// 第二个数字代表g(群组)权限，
	// 第三个数字代表o(其他人)权限
	// 每一个数字都是通过4（表示r）,2（表示w）,1（表示x）三个数字相加得到的
	// rwx 对应4，2，1
	// 那么只读的权限用4表示[r--]，
	// 读写用6（4+2）表示[rw-],
	// 读写加执行用7（4+2+1）表示[rwx]。
	if err != nil{
		fmt.Println("os.OpenFile err:",err)
		return
	}
	// 最终 关闭流
	defer file.Close()
	// 创建写缓冲区
	writer := bufio.NewWriter(file)
	// 取出棋子
	for _, valNode:=range sparseArr{
		str:=fmt.Sprint(valNode.row,valNode.col,valNode.val)
		// 写到缓存区中
		_, err := writer.WriteString(str+"\n")
		if err != nil{
			fmt.Println("writer.WriteString(str) err:",err)
			return
		}
	}
	// 刷新数据, 将缓冲区数据写入io writer
	writer.Flush()
}

func readSparse()  {
	type ValNode struct {
		row int
		col int
		val int
	}
	var sparseArr []ValNode

	filePath := "./数据结构/sparseArr.txt"
	file, err := os.Open(filePath)
	if err != nil {
		fmt.Println("os.Open(filePath) err:", err)
		return
	}
	defer file.Close()
	// 创建读缓冲区
	reader := bufio.NewReader(file)
	// 读取信息
	for{
		// 读取delim之前的所有string数据
		str,err := reader.ReadString('\n')
		if err==io.EOF { 	// io.EOF代表文件的末尾
			break
		}
		fmt.Println(str)
		// 返回将字符串按照空白
		//（unicode.IsSpace确定， 可以是一到多个连续的空白字符）
		// 分割的多个字符串。
		// 如果字符串全部是空白或者是空字符串的话，会返回空切片。
		arr := strings.Fields(str)
		valNode := ValNode{}
		for i, _ := range arr {
			in,err := strconv.Atoi(arr[i])
			if err != nil {
				return
			}
			// arr 中 第一个位置是row 第二个位置是col 第三个位置是val
			switch i {
			case 0:
				valNode.row = in
			case 1:
				valNode.col = in
			case 2:
				valNode.val = in
			}
		}
		sparseArr = append(sparseArr,valNode)
	}
	// 棋盘
	var chessMap [][]int
	//var chessMap [11][11]int
	for i, v := range sparseArr {
		if i == 0 {
			for a := 0; a < v.row; a++ {
				mm := make([]int, v.col)
				chessMap = append(chessMap, mm)
			}
		} else {
			chessMap[v.row][v.col] = v.val
		}

	}
	fmt.Println(chessMap)
}