📅  最后修改于: 2023-12-03 14:57:49.881000             🧑  作者: Mango
本文介绍了如何使用Java编写一个生成身份矩阵的程序。身份矩阵是一个方阵,对角线上的元素全为1,其他元素全为0。该程序可以接受用户输入矩阵的大小,并生成对应的身份矩阵。
public class IdentityMatrixGenerator {
public static void main(String[] args) {
int size = getUserInput();
int[][] matrix = generateIdentityMatrix(size);
printMatrix(matrix);
}
private static int getUserInput() {
// 使用Scanner接收用户输入的矩阵大小
Scanner scanner = new Scanner(System.in);
System.out.print("请输入矩阵的大小:");
return scanner.nextInt();
}
private static int[][] generateIdentityMatrix(int size) {
int[][] matrix = new int[size][size];
for (int i = 0; i < size; i++) {
for (int j = 0; j < size; j++) {
if (i == j) {
matrix[i][j] = 1; // 对角线上的元素设为1
}
}
}
return matrix;
}
private static void printMatrix(int[][] matrix) {
// 遍历矩阵并按行打印
System.out.println("生成的身份矩阵:");
for (int[] row : matrix) {
for (int element : row) {
System.out.print(element + " ");
}
System.out.println();
}
}
}
上述代码包含以下几个部分:
main
方法:程序的入口,调用getUserInput
方法获取用户输入的矩阵大小,并传递给generateIdentityMatrix
方法生成身份矩阵,最后调用printMatrix
方法打印矩阵。getUserInput
方法:使用Scanner
类接收用户输入的矩阵大小,并返回输入的值。generateIdentityMatrix
方法:根据输入的矩阵大小创建一个二维数组,并将对角线上的元素设为1。printMatrix
方法:遍历矩阵并按行打印。输入:
请输入矩阵的大小:5
输出:
生成的身份矩阵:
1 0 0 0 0
0 1 0 0 0
0 0 1 0 0
0 0 0 1 0
0 0 0 0 1
这是一个5x5的身份矩阵,对角线上的元素都是1,其他元素都是0。
以上就是一个生成身份矩阵的Java程序,可以根据用户输入的矩阵大小生成对应的身份矩阵,并将其打印出来。该程序可以灵活地适应不同大小的矩阵。