Go语言学习笔记【11】 数据结构之稀疏矩阵、队列、栈

【声明】

非完全原创，部分内容来自于学习其他人的理论和B站视频。如果有侵权，请联系我，可以立即删除掉。

一、稀疏矩阵

若数值为0的元素数目远远多于非0元素的数目，并且非0元素分布没有规律时，则称该矩阵为稀疏矩阵；与之相反，若非0元素数目占大多数时，则称该矩阵为稠密矩阵。

在存储稀疏矩阵时，可以通过只存储矩阵的行数、列数、默认值，以及非0元素的行列信息，即可达到存储整个矩阵的目的。例如，当围棋中棋子的个数很少的时候，可以通过该方式保存棋局，下一次打开时也方便恢复

1、存储原理

如下图，将一个矩阵中的非0值存储为三元格式的数组，数组的第一行保存稀疏矩阵的行数、列数、默认值；后面开始保存非默认值的行索引、列索引、数值

2、代码实现

package main

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

type Node struct {
   
	row int
	col int
	val int
}

const rows, cols = 6, 7

func Matrix_Conver(arr [rows][cols]int, default_val int) []Node {
   
	var node []Node
	node = append(node, Node{
   rows, cols, default_val})
	for i, ar := range arr {
   
		for j, val := range ar {
   
			if val != default_val {
   
				node = append(node, Node{
   i, j, val})
			}
		}
	}
	return node
}

func PrintMatrix(arr [rows][cols]int) {
   
	for _, ar := range arr {
   
		for _, val := range ar {
   
			fmt.Printf("%d\t", val)
		}
		fmt.Println()
	}
}

func main() {
   
	//1. 构造稀疏矩阵
	def_val := -1
	path := "D:/array.data"
	array := [rows][cols]int{
   
		{
   def_val, def_val, def_val, 2, def_val, def_val, def_val},
		{
   def_val, def_val, def_val, def_val, def_val, def_val, def_val},
		{
   6, def_val, def_val, def_val, def_val, def_val, def_val},
		{
   def_val, def_val, 9, def_val, def_val, def_val, 1},
		{
   def_val, def_val, def_val, def_val, def_val, def_val, def_val},
		{
   def_val, def_val, def_val, def_val, def_val, 8, def_val},
	}
	fmt.Println("original matrix:")
	PrintMatrix(array)
	//2. 转为3元切片保存
	node := Matrix_Conver(array, def_val)
	fmt.Println("\nconversion to slice:")
	fmt.Println(node)
	//3. 保存3元切片到文件中
	fp, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0666)
	if err != nil {
   
		fmt.Println("file open by create and read/write mode failed, err = ", err)
		return
	}
	writer := bufio.NewWriter(fp)
	for _, data := range node {
   
		_, err = writer.WriteString(strconv.Itoa(data.row) + "," + strconv.Itoa(data.col) + "," +
			strconv.Itoa(data.val) + "\n")
		if err != nil {
   
			fmt.Println("writer write string failed, err = ", err)
			return
		}
	}
	writer.Flush() //一定要调用才能写入到文件中
	fp.Close()
	//4. 从文件中读取内容（注意，读出来的是字符格式的数字）
	content, err := ioutil.ReadFile(path)
	if err != nil {
   
		fmt.Println("ioutil.ReadFile failed, err = ", err)
		return
	}
	fmt.Println("\nread all content from file:")
	fmt.Println(string(content))
	//5. 还原为稀疏矩阵
	str := strings.Split(string(content), "\n")
	default_val, err := strconv.Atoi(strings.Split(str[0], ",")[2])
	if err != nil {
   
		fmt.Println("strconv.Atoi failed, err = ", err)
		return
	}
	var matrix [rows][cols]int
	for i := 0; i < len(matrix); i++ {
   
		for j := 0; j < len(matrix[0]); j++ {
   
			matrix[i][j] = default_val
		}
	}
	//注意，最后一行是'\n'，因此str切片的最后一个字符串是空字符串""
	for i := 1; i < len(str)-1; i++ {