本文实例讲述了Python实现螺旋矩阵的填充算法。分享给大家供大家参考,具体如下:afanty的
本文实例讲述了Python实现螺旋矩阵的填充算法。分享给大家供大家参考,具体如下:
afanty的分析:
关于矩阵(二维数组)填充问题自己动手推推,分析下两个下表的移动规律就很容易咯。
对于螺旋矩阵,不管它是什么鬼,反正就是依次向右、向下、向右、向上移动。
向右移动:横坐标不变,纵坐标加1 向下移动:纵坐标不变,横坐标加1 向右移动:横坐标不变,纵坐标减1 向上移动:纵坐标不变,横坐标减1
代码实现:
#coding=utf-8
import numpy
'''''
Author: afanty
Date: 2016/6/23
'''
def helixMatrix(n):
'''''实现n维螺旋矩阵的填充
:param n:维数
:return:螺旋矩阵
'''
if not isinstance(n, int) or n <= 0:
raise ValueError('请输入合适的维数')
matrix = numpy.zeros((n, n))
left_top = 0
right_buttom = n - 1
number = 1
while left_top < right_buttom:
# 向右移动,横坐标不变,纵坐标+1,number+1
i = left_top
while i < right_buttom:
matrix[left_top][i] = number
i += 1
number += 1
# while
# 向下移动,纵坐标不变,横坐标+1,number+1
i = left_top
while i < right_buttom:
matrix[i][right_buttom] = number
i += 1
number += 1
#while
# 向左移动,横坐标不变,纵坐标-1,number+1
i = right_buttom
while i > left_top:
matrix[right_buttom][i] = number
i -= 1
number += 1
# while
# 向上移动,纵坐标不变,横坐标-1,number+1
i = right_buttom
while i > left_top:
matrix[i][left_top] = number
i -= 1
number += 1
# while
left_top += 1
right_buttom -= 1
# while
if n % 2 != 0:
matrix[n / 2][n / 2] = n * n
return matrix
# end
print("脚本之家测试结果:")
print helixMatrix(5)
运行结果:
Python 螺旋矩阵 填充算法