Python3程序生成二次对角线之和等于完美平方的矩阵
给定一个整数N ,任务是使用[1, N]范围内的正整数生成一个维度为N x N的矩阵,使得次对角线的总和是一个完美的正方形。
例子:
Input: N = 3
Output:
1 2 3
2 3 1
3 2 1
Explanation:
The sum of secondary diagonal = 3 + 3 + 3 = 9(= 32).
Input: N = 7
Output:
1 2 3 4 5 6 7
2 3 4 5 6 7 1
3 4 5 6 7 1 2
4 5 6 7 1 2 3
5 6 7 1 2 3 4
6 7 1 2 3 4 5
7 1 2 3 4 5 6
Explanation:
The sum of secondary diagonal = 7 + 7 + 7 + 7 + 7 + 7 + 7 = 49(= 72).
方法:由于生成的矩阵需要具有N x N的维度,因此,为了使辅助对角线中的元素之和成为一个完美的正方形,我们的想法是在辅助对角线的每个索引处分配N。因此,这条对角线上的所有N个元素之和为N 2 ,这是一个完美的正方形。请按照以下步骤解决问题:
- 初始化维度为N x N的矩阵mat[][] 。
- 将矩阵的第一行初始化为 {1 2 3 ... N}。
- 现在对于矩阵的其余行,通过矩阵的前一行的排列的循环左移填充每一行1 。
- 完成上述步骤后打印矩阵。
下面是上述方法的实现:
Python3
# Python3 program for the above approach
# Function to print the matrix whose sum
# of element in secondary diagonal is a
# perfect square
def diagonalSumPerfectSquare(arr, N):
# Print the current row
print(*arr, sep =" ")
# Iterate for next N - 1 rows
for i in range(N-1):
# Perform left shift by 1
arr = arr[i::] + arr[:i:]
# Print the current row after
# the left shift
print(*arr, sep =" ")
# Driver Code
# Given N
N = 7
arr = []
# Fill the array with elements
# ranging from 1 to N
for i in range(1, N + 1):
arr.append(i)
# Function Call
diagonalSumPerfectSquare(arr, N)
输出
1 2 3 4 5 6 7
2 3 4 5 6 7 1
3 4 5 6 7 1 2
4 5 6 7 1 2 3
5 6 7 1 2 3 4
6 7 1 2 3 4 5
7 1 2 3 4 5 6
时间复杂度: O(N 2 )
辅助空间: O(N)
有关更多详细信息,请参阅有关生成具有二次对角线之和等于完美平方的矩阵的完整文章!