二维数组转稀疏数组、稀疏数组恢复二维数组(Java实现)

public static void main(String[] args) {
    // 创建一个原始的二维数组 9*9
    int chessArr1[][] = new int[9][9];
    // 0表示没有棋子，1表示黑子，2表示白子
    chessArr1[1][2] = 1;
    chessArr1[2][3] = 2;
    chessArr1[5][2] = 2;
    chessArr1[7][6] = 2;
    chessArr1[8][4] = 1;
    System.out.println("原始的二维数组");
    for (int[] row : chessArr1) {
        for (int data : row) {
            System.out.printf("%d\t", data);
        }
        System.out.println();
    }
    
    // 将二维数组转为稀疏数组
    // 1. 先遍历二维数组
    int sum = 0;
    for (int[] row : chessArr1) {
        for (int data : row) {
            sum += data != 0 ? 1 : 0;
        }
    }
    System.out.println("一共有" + sum + "个值");
    // 2. 创建对应的稀疏数组
    int sparseArr[][] = new int[sum + 1][3];
    // 3.给稀疏数组第一行赋值
    sparseArr[0][0] = 9;
    sparseArr[0][1] = 9;
    sparseArr[0][2] = sum;
    // 4.遍历二维数组，给稀疏数组其他行赋值
    int cur = 0;
    for (int i = 0; i < 9; i++) {
        for (int j = 0; j < 9; j++) {
            if (chessArr1[i][j] != 0) {
                cur++;
                sparseArr[cur][0] = i;
                sparseArr[cur][1] = j;
                sparseArr[cur][2] = chessArr1[i][j];
            }
        }
    }
    System.out.println("转换后的稀疏数组");
    for (int[] row : sparseArr) {
        for (int data : row) {
            System.out.printf("%d\t", data);
        }
        System.out.println();
    }

    // 将稀疏数组恢复为二维数组
    int x = sparseArr[0][0];
    int y = sparseArr[0][1];
    int count = sparseArr[0][2];
    int chessArr2[][] = new int[x][y];
    for (int i = 1; i < count + 1; i++) {
        chessArr2[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
    }
    System.out.println("恢复的二维数组");
    for (int[] row : chessArr1) {
        for (int data : row) {
            System.out.printf("%d\t", data);
        }
        System.out.println();
    }
}