数据结构之路：稀疏数组

假如，我们有一个二维数组array，里面大部分的数据都是0，只有两个数字不一样，分别是array[1][2] = 1 和 array[2][3] = 2。
[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]

如果我们要把这个二维数组保存到文件的话，需要保存11 * 11 = 121条数据，但是如果我们使用稀疏数组的话，可以大大减少保存的数据。

1. 稀疏数组

稀疏数组是一个N + 1行，3列的二维数组。
第一行的数据分别为：二维数组的总行数、二维数组的总列数、二维数组不同值的个数。
从第二行开始的数据分别为：元素的行坐标、元素的列坐标、元素的值。

现在，我们把刚开始展示的那个二维数组转换为稀疏数组。
[11, 11, 2]
[1, 2, 1]
[2, 3, 2]

第一行的[11, 11, 2]表示原来的二维数组有11行、11列、2个不同的值
第二行的[1, 2, 1]表示原来的二维数组array[1][2] = 1
第三行的[2, 3, 2]表示原来的二维数组array[2][3] = 2

2. 二维数组转换为稀疏数组

private static int[][] array2SparseArray(int[][] array) {
    int diffValueCount = 0;
    for (int[] row : array) {
        for (int value : row) {
            // 这个地方我们默认二维数组的大部分值为0
            if (value != 0) {
                diffValueCount++;
            }
        }
    }
    int[][] sparseArray = new int[diffValueCount + 1][3];
    sparseArray[0][0] = array.length;
    sparseArray[0][1] = array[0].length;
    sparseArray[0][2] = diffValueCount;
    int index = 1;
    for (int i = 0; i < array.length; i++) {
        for (int j = 0; j < array[i].length; j++) {
            if (array[i][j] != 0) {
                sparseArray[index][0] = i;
                sparseArray[index][1] = j;
                sparseArray[index][2] = array[i][j];
                index++;
            }
        }
    }
    return sparseArray;
}

3. 稀疏数组转换为二维数组

private static int[][] sparseArray2Array(int[][] sparseArray) {
    int[][] array = new int[sparseArray[0][0]][sparseArray[0][1]];

    for (int i = 1; i < sparseArray.length; i++) {
        array[sparseArray[i][0]][sparseArray[i][1]] = sparseArray[i][2];
    }
    return array;
}

4. 保存稀疏数组到文件

private static void saveToFile(int[][] array) {
    try {
        FileWriter fw = new FileWriter("./array.data");
        for (int i = 0; i < array.length; i++) {
            for (int j = 0; j < array[i].length; j++) {
                fw.write(array[i][j] + " ");
            }
            fw.write("\n");
        }
        fw.close();
    } catch (IOException e) {
        e.printStackTrace();
    }
}

5. 从文件中读取稀疏数组

private static int[][] getArrayFromFile() {
    int[][] array = null;
    int index = 0;
    try {
        FileReader fr = new FileReader("./array.data");
        BufferedReader br = new BufferedReader(fr);
        String row;
        while ((row = br.readLine()) != null) {
            String[] data = row.split(" ");
            if (array == null) {
                array = new int[Integer.valueOf(data[2]) + 1][3];
            }
            array[index][0] = Integer.parseInt(data[0]);
            array[index][1] = Integer.parseInt(data[1]);
            array[index][2] = Integer.parseInt(data[2]);
            index++;
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
    return array;
}