![46e6c49b9e8c5ada1486581fe44dc4e3.png](https://img-blog.csdnimg.cn/img_convert/46e6c49b9e8c5ada1486581fe44dc4e3.png)
点击上方「蓝字」关注我们
![7fbe92298ed53093b2a30ce9efb2f078.gif](https://img-blog.csdnimg.cn/img_convert/7fbe92298ed53093b2a30ce9efb2f078.gif)
题目描述:“给定一个含有M*N个元素的矩阵,请以对角线遍历的顺序返回矩阵中的元素”
(矩阵中元素<=10000)
示例:
matrix=[[1, 2, 3],[4, 5, 6],[7, 8, 9],]
[1, 2, 4, 7, 5, 3, 6, 8, 9]
算法思路
寻找元素遍历的规律
按照对角线将元素划分,你会发现一些规律:
- 将每一个对角线上的元素各自看成一组,对角线数组的个数如下:(len(matrix)+len(matrix[0])-1)
- 如果将对角线数组从左向右数,你会发现第偶数个对角线数组在读取时应该逆序读取
- 最终要的规律:在同一个对角线数组中的元素下标之后相等
算法步骤:
- 先按照矩阵遍历的正常顺序遍历元素,过程中计算元素下标和
- 将下标和一样的元素存入对应的对角线数组中(设置一个二维数组存储对角线数组)同时要注意:二维数组中每一个一维数组的对应下标其实正好就等于之前计算的元素下标记和
- 遍历完成后,将对角线二维数组中的一维数组再添加到函数最后要返回的数组中。其中要进行判断:第偶数个数组要逆序读取
本文由“壹伴编辑器”提供技术支持
注意:
- 一定要判断输入为空的情况
- 输入可没说一定是方阵,所以在自己举例推导时不要取特殊情况
- 既然所有元素最终都要返回,那就先把每个都考虑进单个数组,最后再排序
本文由“壹伴编辑器”提供技术支持
二维数组的创建问题
创建二维数组的坑
本文题目在之前的算法中一直这样创建二维数组:
diagonal_list = [[]*(len(matrix)+len(matrix[0]-1)]
但是一直报错
原来二维数组的创建有坑!!!!!!!!
a = [ []*3 ]a[0].append(1)>>> a=[ [1], [1], [1] ]
都变成1了,可明明仅设置了第一个数组呀?????
原来:在创建一个如果利用乘法创建二维数组的话,或者不是二维数组,仅仅是一维数组乘以n,乘法代表的是引用,后面不论多少个数组都是引用第一个数组
本文由“壹伴编辑器”提供技术支持
正确的创建方法:利用for循环
a = [] for i in range(n)