Java实现构建稀疏数组

package com.wy.sparsearray;
import java.util.Scanner;
/**
 * @author HelloWorld
 * @create 2021-03-30-21:06
 * @email 154803771@qq.com
 */
public class SparseArray {
    public static void main(String[] args){
        //先创建一个原始的二维数组
        Scanner sc = new Scanner(System.in);
        System.out.println("输入初始二维数组的行");
        int arrayRows = sc.nextInt();
        System.out.println("输入初始二维数组的列");
        int arrayLines = sc.nextInt();
        int[][]  twoDimensionalArray = new int[arrayRows][arrayLines];
        //对初始二维数组进行赋值，未赋值部分默认值为0
        int flag = 0;
        do {
            System.out.println("待赋值处所在行[>=1]");
            int arrayRow = sc.nextInt();
            System.out.println("待赋值处所在列[>=1]");
            int arrayLine = sc.nextInt();
            System.out.println("赋值处的值[int]");
            int value = sc.nextInt();
            System.out.println("是否继续输入 1/0");
            twoDimensionalArray[arrayRow - 1][arrayLine - 1] = value;
            flag = sc.nextInt();
        } while (flag == 1);

        //输出原始数组
        //增强for循环
        System.out.println("初始二维数组>>>>>>>>>>>>>>>>>>");
        for(int[] row : twoDimensionalArray){
            for(int data : row){
                System.out.print("\t" + data);
            }
            System.out.println();
        }

        //将二维数组转换为稀疏数组
        //1. 先遍历二维数组 得到非0数据的个数
        int sum = 0;
        for (int i = 0; i < arrayRows; i++) {
            for (int j = 0; j < arrayLines; j++) {
                if (twoDimensionalArray[i][j] != 0){
                    sum++;
                }
            }
        }
        //2. 创建对应的稀疏数组
        int[][] sparseArray = new int[sum + 1][3];
        //给稀疏数组赋值
        sparseArray[0][0] = arrayRows;
        sparseArray[0][1] = arrayLines;
        sparseArray[0][2] = sum;

        // 遍历二维数组将非0值存放到稀疏数组
        int count = 0;  //用于记录是第几个在稀疏数组中的数据
        for (int i = 0; i < arrayRows; i++) {
            for (int j = 0; j < arrayLines; j++) {
                if (twoDimensionalArray[i][j] != 0){
                    count++;
                    sparseArray[count][0] = i;
                    sparseArray[count][1] = j;
                    sparseArray[count][2] = twoDimensionalArray[i][j];
                }
            }
        }

        //输出稀疏数组的形式
        System.out.println();
        System.out.println("得到的稀疏数组>>>>>>>>>>>>>>>>>>");
        for(int[] row : sparseArray){
            for(int data : row){
                System.out.print("\t" + data);
            }
            System.out.println();
        }

        //将稀疏数组恢复成原始的二维数组
        // 1. 构建二维数组
        int[][] array= new int[sparseArray[0][0]][sparseArray[0][1]];

        // 2.赋值
        for (int i = 1; i <= sparseArray[0][2]; i++) {
            array[sparseArray[i][0]][sparseArray[i][1]] = sparseArray[i][2];
        }
        // 输出还原后的二维数组
        System.out.println();
        System.out.println("还原后的二维数组>>>>>>>>>>>>>>>>>>");
        for(int[] row : array){
            for(int data : row){
                System.out.print("\t" + data);
            }
            System.out.println();
        }
    }
}