用对角线去遍历矩阵

声明

该系列文章仅仅展示个人的解题思路和分析过程,并非一定是优质题解,重要的是通过分析和解决问题能让我们逐渐熟练和成长,从新手到大佬离不开一个磨练的过程,加油!

原题链接

用对角线遍历矩阵https://leetcode.cn/leetbook/read/array-and-string/cuxq3/

算法分析

图一

图二

图三
图四

由上述四个图可以总结得出以下八个结论:

结论1: k属于[0,a(max)+b(max)]。

结论2: 每一层遍历行最多存在min(m,n)个矩阵索引对,min(m,n)表示m和n二者中小的那一个值。

结论3: a属于[0,m-1],b属于[0,n-1]。

结论4: 若k为偶数则以a为比较对象,此时若k<=a(max)则当前遍历行的起始索引对为(k,0),反之则起始索引对为(a(max),k-a(max))。

结论5: 若k为奇数则以b为比较对象,此时若k<=b(max)则当前遍历行的起始索引对位(0,k),反之则起始索引对为(k-b(max),b(max))。

结论6: 从当前遍历行的起始索引对开始,若k为偶数,则下一个索引对为(a-1,b+1),反之下一个索引对为(a+1,b-1)。

结论7: 遍历行的结束条件由该矩阵的遍历行最大索引对个数和a(min)、b(min)共同决定,首先应判断当前遍历行访问的矩阵索引对个数是否达到了该矩阵的遍历行最大索引对个数,若达到了则完成该遍历行的遍历,否则判断下一个索引对(a,b),若a或b越界则表明完成该遍历行的遍历。

结论8: 整个矩阵遍历结束的条件是当前遍历的矩阵索引对的个数是m×n个。

代码示例(C#)
cs 复制代码
public int[] FindDiagonalOrder(int[][] mat)
{
    int m = mat.Length;//矩阵的行数
    int n = mat[0].Length;//矩阵的列数
    int[] result = new int[m * n];//结果数组
    //a:索引对的a,b:索引对的b,k:a+b,count:当前已遍历的矩阵索引对个数,minA:a的最小值
    //minB:b的最小值,index:结果数组中的索引指针
    int a, b, k = 0, count = 0, minA = 0, minB = 0, index = 0;
    int maxA = m - 1;//a的最大值
    int maxB = n - 1;//b的最大值
    int maxIndexsCount = m <= n ? m : n;//该矩阵中遍历行的矩阵索引对个数的最大值
    int lineIndexsCount;//遍历行当前已遍历的矩阵索引对个数
    while (count < m * n)
    {
        lineIndexsCount = 0;
        //若k%2为0则表示k为偶数,反之则为奇数
        if (k % 2 == 0)
        {
            if (k <= maxA)
            {
                a = k;
                b = 0;
            }
            else
            {
                a = maxA;
                b = k - maxA;
            }
        }
        else
        {
            if (k <= maxB)
            {
                a = 0;
                b = k;
            }
            else
            {
                a = k - maxB;
                b = maxB;
            }
        }
        while (lineIndexsCount < maxIndexsCount)
        {
            //a或b越界则完成此次遍历行的遍历
            if ((a < minA || a > maxA) || (b < minB || b > maxB)) break;
            //记录结果
            result[index++] = mat[a][b];
            count++;
            if (k % 2 == 0)
            {
                a--;
                b++;
            }
            else
            {
                a++;
                b--;
            }
        }
        k++;
    }
    return result;
}
算法解说

按照原题的思路我们列举出了四种矩阵,严格来说只有三种,分别是矩阵行数大于列数、行数等于列数、行数小于列数,而为了保证后续结论的真实性,我们列举了两个行数等于列数的矩阵。

图中我们完成了四个矩阵的制定,然后按照题目要求去遍历了四个矩阵并且将遍历的结果记录下来,从行列定义上我们可以将矩阵构建在二维平面中,从而为每一个元素设立一个唯一的坐标值,在本题中我们把这个坐标值称为索引对。有了索引对我们可以进一步按照索引对两个数值之和对遍历结果进行分组,在本题中我们把索引对两个数值之和相同的一组索引对称为遍历行,为了简便后续将索引对两个数值之和称为索引对结果值。

将遍历行按照索引对结果值从小到大的顺序进行排列,为了能够发现更多有用的结论,我们把每一层遍历行的索引对结果值、结果值的奇偶性、索引对与k值的关系列举出来,除此之外还需要列举出矩阵行列数以及索引对两个数值的取值范围,至于为什么要去列举这些东西,实际上还是通过尝试和思考,就像上学时做数学题,浏览题目之后列举出题目中给定的条件,然后根据条件去解题。

这一步我们可以理解为进一步细化和剖析已知条件,我们的目的是从已知条件中获取可靠的结论,从而根据结论解决问题,因此我们总结出了上文的八个结论。

那么已知结论后如何去编写代码呢?本人是按照以下几个问题去解决的。

1.我们需要明白输入和输出是什么?

2.我们需要定义哪些变量?

3.函数主体是什么?

4.某些过程的结束条件是什么?

实际上,就是将我们的八个结论构建为代码,简单划分就是定义变量和逻辑主体,定义变量就看结论中需要记录数据的描述,逻辑主体就看结论中对于逻辑处理、结束条件的描述。

当然这样转换出来的代码比较粗糙,所以后续还可以结合自己的编程知识再去优化自己的代码,让它变得更加简洁精致。

相关推荐
passer__jw76727 分钟前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾34 分钟前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序42 分钟前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
爱吃生蚝的于勒1 小时前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~1 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^2 小时前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城2 小时前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德2 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz2 小时前
STL--哈希
c++·算法·哈希算法
kingmax542120082 小时前
初三数学,最优解问题
算法