Python 生成蛇形数组/对角线数组

最新推荐文章于 2024-05-08 13:33:12 发布

原创最新推荐文章于 2024-05-08 13:33:12 发布 · 2.1k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法

Python 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了如何使用Python生成蛇形和对角线填充的二维数组。通过理解数组中对角线元素的下标规律，分别展示了对角线填充时的直接填充方法和蛇形填充时方向交替的实现策略。提供了相应的代码示例。

部署运行你感兴趣的模型镜像

问题描述

生成一个m * n的二维数组，元素值为1，2，3…按对角线顺序由下向上排列，如：

1，3，6，9
2，5，8，11
4，7，10，12

生成一个m * n的二维数组，元素值为1，2，3…按蛇形顺序排列，如：

1，2，6，7
3，5，8，11
4，9，10，12

思路

数组同一条斜线上的元素，其下标和相等，从0到m+n-2递增。

遍历0到m+n-2下标和，每次填入一条对角线上的值。

对角线数组方向一致，很容易生成；

蛇形数组交替改变方向，可将每两次循环视为一组。

代码

对角线数组：

import numpy as np
m = 3 #行数
n = 4 #列数
list = np.zeros((m, n), dtype=np.int)
list[0][1] = 1
count = 1
#k记录下标和。每个斜线上下标和相等，从0到m+n-2递增
for k in range(m+n-1):
    for i in range(max(k-m+1, 0), min(k+1, n)):
        list[k-i][i] = count
        count += 1
print(list)

蛇形数组：

import numpy as np
m = 3 #行数
n = 4 #列数
list = np.zeros((m, n), dtype=np.int)
count = 1
k = 0 #记录下标和。每个斜线上下标和相等，从0到m+n-2递增
while k<m+n-1:
    for i in range(max(k-m+1, 0), min(k+1, n)):
        list[k-i][i] = count
        count += 1
    k += 1
    if k<m+n-1:
        for i in range(max(k-n+1, 0), min(k+1, m)):
            list[i][k-i] = count
            count += 1
    k += 1
print(list)