【leetcode 498. 对角线遍历】Diagonal Traverse

最新推荐文章于 2022-09-04 20:36:32 发布

野人Boom

最新推荐文章于 2022-09-04 20:36:32 发布

阅读量526

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode C

本文链接：https://blog.youkuaiyun.com/TNTcainiao/article/details/82846239

本文介绍了一种矩阵对角线遍历的算法实现，通过分析矩阵元素的坐标规律，采用左下到左上和右上到右下的遍历方式，实现了矩阵元素的有效遍历。适用于MxN矩阵，总元素不超过100000。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个含有 M x N 个元素的矩阵（M行，N列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

示例:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出:  [1,2,4,7,5,3,6,8,9]
解释:

说明:

给定矩阵中的元素总数不会超过 100000 。

分析题目，遍历分左下到左上和右上到右下

观察每个对角线上的元素发现在同一对角线上元素的坐标和为一定值，如第一的对角线为2，依次为3，4，5，6。最大到行数和列数之和。

故可从2开始到行数和列数和遍历。每一个值对应一条对角线，行依次减，列依次对应加为左下到左上。行依次加，列依次减对应右上到右下，注意边界条件即可。

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findDiagonalOrder(int** matrix, int matrixRowSize, int matrixColSize, int* returnSize) {
    *returnSize = matrixRowSize * matrixColSize;  
    int size = matrixRowSize + matrixColSize;
    int bit = 0,row,col,state=0;
    int *obj = (int*)malloc((*returnSize) * sizeof(int));
    
    for(int i=2;i<=size;i++){       
        if(bit==0){
            if(i>matrixRowSize + 1) col = i - matrixRowSize;
            else col = 1;
            
            for(int j=col;j<i && j-1 < matrixColSize && i-j-1<matrixRowSize;j++){
                obj[state] = matrix[i-j-1][j-1];
                state++;               
            }           
            bit = 1;
            continue;
        }
        if(bit==1){
             if(i>matrixColSize + 1) row = i - matrixColSize;
            else row = 1;
            
             for(int j=row;j<i && j-1<matrixRowSize && i-j-1<matrixColSize;j++){
                 obj[state] = matrix[j-1][i-j-1];
                state++;                
            }
            
            bit = 0; 
        }
    }   
    return obj;
    
}