【Java数据结构】1.稀疏矩阵

package com.gdupt.encapsulation;

import java.io.*;
import java.util.ArrayList;

/**
 * @author Artemis
 * @version 1.0
 *
   版权声明：本文为CSDN博主「ArtemisKhryso」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
   原文链接：https://blog.csdn.net/ArtemisKhryso/article/details/117384988
 */
public class SparseArray {


	public static void main(String[] args) throws IOException {
		// 创建一个原始的二维数组 11 * 11
		// 0: 表示没有棋子， 1 表示 黑子 2 表蓝子
		int binArr[][] = new int[11][11];
		binArr[1][2] = 1;
		binArr[2][3] = 2;
		binArr[4][5] = 2;

		// 输出原始的二维数组
		System.out.println("原始二维数组：");
		showArr(binArr);
		System.out.println("----------");

		//生成稀疏数组
		int[][] sparseArr = binarrToSparse(binArr, 11, 11);
		System.out.println("稀疏数组：");
		showArr(sparseArr);
		System.out.println("----------");

		//保存稀疏数组到磁盘再读取
		outSparseArray(sparseArr, "sparse.txt");
		int sparseArr2[][] = inSparseArray("sparse.txt");
		System.out.println("磁盘读取的稀疏数组：");
		showArr(sparseArr2);
		System.out.println("----------");

		//读取的稀疏数组转二维数组
		int binArr2[][] = sparseToBinarr(sparseArr2);
		System.out.println("复原后二维数组：");
		showArr(binArr2);
		System.out.println("----------");

	}

	/**
	 * 将原始的二维数组binarr压缩成稀疏数组sparse
	 * @param binArr
	 * @return
	 */
	public static int[][] binarrToSparse(int[][] binArr, int row, int col){
		// 1. 先遍历二维数组 得到非0数据的个数
		int sum = 0;
		for (int i = 0; i < row; i++) {
			for (int j = 0; j < col; j++) {
				if (binArr[i][j] != 0) {
					sum++;
				}
			}
		}

		// 2. 创建对应的稀疏数组
		int sparseArr[][] = new int[sum + 1][3];
		// 给稀疏数组赋值
		sparseArr[0][0] = row;
		sparseArr[0][1] = col;
		sparseArr[0][2] = sum;

		// 遍历二维数组，将非0的值存放到 sparseArr中
		int count = 0; //count 用于记录是第几个非0数据
		for (int i = 0; i < row; i++) {
			for (int j = 0; j < col; j++) {
				if (binArr[i][j] != 0) {
					count++;
					sparseArr[count][0] = i;
					sparseArr[count][1] = j;
					sparseArr[count][2] = binArr[i][j];
				}
			}
		}

		return sparseArr;
	}

	/**
	 * 将稀疏数组sparse恢复成原始的二维数组binarr
	 * @param sparseArr
	 * @return
	 */
	public static int[][] sparseToBinarr(int[][] sparseArr){
		//1. 先读取稀疏数组的第一行，根据第一行的数据，创建原始的二维数组
		int binArr[][] = new int[sparseArr[0][0]][sparseArr[0][1]];

		//2. 在读取稀疏数组后几行的数据(从第二行开始)，并赋给 原始的二维数组 即可
		for(int i = 1; i < sparseArr.length; i++) {
			binArr[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
		}

		return binArr;
	}

	/**
	 * 稀疏数组输出保存
	 * @param arr
	 * @throws IOException
	 */
	public static void outSparseArray(int[][] arr, String outPatn) throws IOException {
		BufferedWriter bw = new BufferedWriter(new FileWriter(outPatn));

		//稀疏数组按个(转成string)写出，也可以按行写出
		for (int[] ints : arr) {
			for (int anInt : ints) {
				bw.write(anInt + "\t");   //write一次,是将数据装到字符数组,装满后再一起写出去
			}
			bw.write("\r\n");
		}

		//关闭资源
		bw.close();
	}

	/**
	 * 稀疏数组读取
	 * @return
	 * @throws IOException
	 */
	public static int[][] inSparseArray(String inPath) throws IOException {
		BufferedReader br = new BufferedReader(new FileReader(inPath));

		//将稀疏矩阵按行放入集合
		ArrayList<String[]> list = new ArrayList<>();
		String line;
		while((line = br.readLine()) != null) {
			list.add(line.split("\t"));
		}

		//取出有效元素个数制造稀疏数组
		int sum = Integer.parseInt(list.get(0)[2]);
		int[][] sparseArr = new int[sum+1][3];

		// 给稀疏数组赋值
		for (int i = 0; i < list.size(); i++) {
			for (int j = 0; j < 3; j++) {
				sparseArr[i][j] = Integer.parseInt(list.get(i)[j]);
			}
		}

		//关闭资源
		br.close();
		return sparseArr;
	}

	/**
	 * 数组展示
	 * @param arr
	 */
	public static void showArr(int[][] arr){
		for (int[] row : arr) {
			for (int cell : row) {
				System.out.print(cell + "\t");
			}
			System.out.println("");
		}
	}
}