Java实现稀疏数组

package fun.xinghai.sparsearray;

public class SparseArray {
	public static void main(String[] args) {
		int[][] array1 = generateArray(10, 10); //生成二维数组
		printArray(array1); //打印二维数组
		int[][] sparseArray = toSparseArray(array1); //二维数组转稀疏数组
		printSparseArray(sparseArray); //打印稀疏数组
		int[][] array2 = toArray(sparseArray); //稀疏数组转二维数组
		printArray(array2); //打印二维数组
		isEquals(array1, array2); //比较两个二维数组是否相等
	}

	/**
	 * 二维数组转稀疏数组
	 */
	public static int[][] toSparseArray(int[][] array) {
		// 1.遍历二维数组，找出元素个数num
		int num = 0;
		for (int[] row : array) {
			for (int elem : row) {
				if (elem != 0) {
					num++;
				}
			}
		}

		// 2.根据num创建稀疏数组，并初始化
		int[][] sparseArray = new int[num + 1][3];
		sparseArray[0][0] = array.length; // 第0行0列存二维数组行数
		sparseArray[0][1] = array[0].length; // 第0行1列存二维数组列数
		sparseArray[0][2] = num; // 第0行2列存二维数组元素个数

		// 3.遍历二维数组，将元素信息填入稀疏数组中
		int count = 0; // 记录元素信息需要填入到稀疏数组中的哪一行
		for (int i = 0; i < array.length; i++) {
			for (int j = 0; j < array[0].length; j++) {
				if (array[i][j] != 0) {
					count++;
					sparseArray[count][0] = i;
					sparseArray[count][1] = j;
					sparseArray[count][2] = array[i][j];
				}
			}
		}
		return sparseArray;
	}

	/**
	 * 稀疏数组转二维数组
	 */
	public static int[][] toArray(int[][] sparseArray) {
		//1.根据稀疏数组的第0行创建二维数组
		int[][] array = new int[sparseArray[0][0]][sparseArray[0][1]];
		
		//2.遍历稀疏数组，根据元素信息填入二维数组中
		for(int i = 1; i < sparseArray.length; i++) {
			array[sparseArray[i][0]][sparseArray[i][1]] = sparseArray[i][2];
		}
		return array;
	}

	/**
	 * 生成二维数组
	 */
	public static int[][] generateArray(int row, int col) {
		int[][] array = new int[row][col];
		for (int i = 0; i < row; i++) {
			array[i][(int) (Math.random() * col)] = (int) (Math.random() * col);
		}
		return array;
	}

	/**
	 * 打印二维数组
	 */
	public static void printArray(int[][] array) {
		System.out.println("二维数组：");
		for (int[] row : array) {
			for (int elem : row) {
				System.out.printf("%d\t", elem);
			}
			System.out.println();
		}
	}

	/**
	 * 打印稀疏数组
	 */
	public static void printSparseArray(int[][] sparseArray) {
		System.out.println("稀疏数组：");
		System.out.println("\trow\tcol\tvalue");
		for (int i = 0; i < sparseArray.length; i++) {
			System.out.printf("%d\t%d\t%d\t%d\n", i, sparseArray[i][0], sparseArray[i][1], sparseArray[i][2]);
		}
	}
	
	/**
	 * 比较两个二维数组是否相等
	 */
	public static void isEquals(int[][] array1, int[][] array2) {
		if(array1.length != array2.length || array1[0].length != array2[0].length) {
			System.out.println("These two arrays are not equal!");
		}else {
			for(int i = 0; i < array1.length; i++) {
				for(int j = 0; j < array1[0].length; j++) {
					if(array1[i][j] != array2[i][j]) {
						System.out.println("These two arrays are not equal!");
						return;
					}
				}
			}
			System.out.println("These two arrays are equal!");
		}
	}
}