稀疏数组

package main

import "fmt"

// main 五子棋棋盘存档-稀疏数组实现存储
func main() {
	// 先定义一个11*11的二维数组，0表示没有棋子，1-表示黑子，2表示白子
	var chessArr [11][11]int
	// 将第二行的第三列设置为黑子，第三行第五列设置为白子
	chessArr[1][2] = 1
	chessArr[2][3] = 2

	// 输出二维棋盘
	for i := 0; i < len(chessArr); i++ {
		for j := 0; j < len(chessArr[i]); j++ {
			fmt.Printf("%d\t", chessArr[i][j])
		}
		fmt.Println()
	}

	// 二维数组转稀疏数组
	// 1、先遍历棋盘，将有棋孽数组计数
	var count int
	for i := 0; i < len(chessArr); i++ {
		for j := 0; j < len(chessArr[i]); j++ {
			if chessArr[i][j] != 0 {
				count++
			}
		}
	}

	// 2. 创建一个稀疏数组（切片）
	var sparseArr = make([][3]int, count+1)
	// 3.将数据存入第一行
	sparseArr[0][0] = len(chessArr)    // 存储原始数组的行数
	sparseArr[0][1] = len(chessArr[0]) // 存储原始数组的列数
	sparseArr[0][2] = count            // 存储非零元素的个数

	// 4. 将数据存入第一行放入
	var row int
	for i := 0; i < len(chessArr); i++ {
		for j := 0; j < len(chessArr[i]); j++ {
			if chessArr[i][j] != 0 {
				row++
				sparseArr[row][0] = i
				sparseArr[row][1] = j
				sparseArr[row][2] = chessArr[i][j]
			}
		}
	}
	// 5. 输出稀疏数组sparseArr
	for i := 0; i < len(sparseArr); i++ {
		fmt.Println(sparseArr[i])
	}

	// 6.根据稀疏数组还原二维数组
	chessArr2 := make([][]int, sparseArr[0][0])
	for i := 0; i < sparseArr[0][0]; i++ {
		chessArr2[i] = make([]int, sparseArr[0][1])
	}
	for i := 1; i < len(sparseArr); i++ {
		chessArr2[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2]
	}
	for _, row := range chessArr2 {
		for _, value := range row {
			fmt.Printf("%d\t", value)
		}
		fmt.Println()
	}
}