编一个程序,定义一个n行n列的二维整数数组,赋初值,然后求出对角线上的元素之和
时间: 2024-10-22 20:16:13 浏览: 26
当你需要编写一个程序来计算给定二维数组的对角线元素之和,你可以采用以下步骤。这里我将以Python为例来解释:
```python
def sum_diagonal(arr, n):
# 初始化对角线元素总和为0
diagonal_sum = 0
# 对于每行
for i in range(n):
# 如果索引合法,将当前行对应的对角线元素加到总和上
if i < n:
diagonal_sum += arr[i][i]
# 如果是下斜对角线,需要从右上角开始
if i >= n - 1 and i + (n - 1) < len(arr):
diagonal_sum += arr[i][len(arr) - i - 1]
return diagonal_sum
# 定义一个n行n列的二维数组,例如n=5
n = 5
array = [[0] * n for _ in range(n)]
# 赋初值,这里只是示例,可以替换成具体的数值
for i in range(n):
for j in range(n):
array[i][j] = i + j # 这里假设数组的每个元素是对角线上距离原点距离的值
# 求对角线元素之和
diagonal_sum = sum_diagonal(array, n)
print("对角线元素之和为:", diagonal_sum)
阅读全文