编程生成一个N阶矩阵,使其主对角线相邻上元素皆为1,与主对角线左侧相邻元素皆为2,其余元素皆为0.
时间: 2024-03-07 15:47:36 浏览: 18
以下是生成 N 阶矩阵的 Python 代码实现:
```python
n = int(input("请输入矩阵阶数:"))
matrix = [[0] * n for _ in range(n)] # 初始化矩阵
# 生成矩阵
for i in range(n):
matrix[i][i] = 1 # 主对角线为1
if i > 0:
matrix[i][i-1] = 2 # 主对角线相邻上元素为2
if i < n-1:
matrix[i][i+1] = 2 # 主对角线相邻下元素为2
# 输出矩阵
for i in range(n):
for j in range(n):
print(matrix[i][j], end=" ")
print()
```
该代码实现了生成 N 阶矩阵的功能,其中主对角线上的元素为1,主对角线相邻上下的元素为2,其余元素为0。时间复杂度为 O(N^2),空间复杂度为 O(N^2)。
相关问题
编程生成一个N阶矩阵,使其主对角线相邻上元素皆为1,与主对角线左侧相邻元素皆为2,其余元素皆为0.c语言
以下是生成N阶矩阵的C语言代码:
```c
#include <stdio.h>
int main() {
int n;
printf("请输入矩阵的阶数:");
scanf("%d", &n);
int matrix[n][n];
int i, j;
for (i = 0; i < n; i++) {
for (j = 0; j < n; j++) {
if (i == j) {
matrix[i][j] = 1;
} else if (j == i - 1) {
matrix[i][j] = 2;
} else if (j == i + 1) {
matrix[i][j] = 2;
} else {
matrix[i][j] = 0;
}
}
}
for (i = 0; i < n; i++) {
for (j = 0; j < n; j++) {
printf("%d ", matrix[i][j]);
}
printf("\n");
}
return 0;
}
```
代码解析:
首先,用户输入矩阵的阶数。
然后,我们创建一个n行n列的二维数组。
接着,我们利用两个for循环,将主对角线上的元素设为1,将主对角线上下相邻的元素设为2。
最后,我们再利用两个for循环,输出生成的矩阵。
举个例子,当n=4时,生成的矩阵为:
```
1 2 0 0
2 1 2 0
0 2 1 2
0 0 2 1
```
编程生成一个N阶矩阵,使其主对角线相邻上元素皆为1,与主对角线左侧相邻元素皆为2,其余元素皆为0.java
以下是 Java 代码实现:
```java
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
System.out.print("请输入矩阵的阶数:");
int n = scanner.nextInt();
// 生成空矩阵
int[][] matrix = new int[n][n];
// 填充主对角线
for (int i = 0; i < n; i++) {
matrix[i][i] = 1;
if (i > 0) {
matrix[i][i-1] = 2;
}
if (i < n-1) {
matrix[i][i+1] = 2;
}
}
// 输出矩阵
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
System.out.print(matrix[i][j] + " ");
}
System.out.println();
}
}
}
```
例如当输入矩阵的阶数为5时,输出的矩阵为:
```
1 2 0 0 0
2 1 2 0 0
0 2 1 2 0
0 0 2 1 2
0 0 0 2 1
```