(LeetCode 每日一题) 498. 对角线遍历 (矩阵、模拟)

题目:498. 对角线遍历

思路:矩阵模拟,时间复杂度0(nm)。
按对角线遍历,而每条对角线都是i+j=k,k的范围为[0,n-1+m-1]。

C++版本:

cpp 复制代码
class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
        int n=mat.size(),m=mat[0].size();
        vector<int> ans;
        // i+j=k  -> j=k-i;
        for(int k=0;k<n+m-1;k++){
        	//枚举j的值,那需要求出j可达的最小值和最大值
        	//当i=n-1时,j的值最小,但不能小于0
            int mn=max(k-(n-1),0);
            //当i=0时,j的值最大,但不能大于m-1
            int mx=min(k-0,m-1);
            if(k%2==0){
                for(int j=mn;j<=mx;j++){
                    ans.push_back(mat[k-j][j]);
                }
            }else{
                for(int j=mx;j>=mn;j--){
                    ans.push_back(mat[k-j][j]);
                }
            }
        }
        return ans;
    }
};

JAVA版本:

java 复制代码
class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int n=mat.length,m=mat[0].length;
        int[] ans=new int[n*m];
        int idx=0;
        // i+j=k  -> j=k-i;
        for(int k=0;k<n+m-1;k++){
            int mn=Math.max(k-(n-1),0);
            int mx=Math.min(k-0,m-1);
            if(k%2==0){
                for(int j=mn;j<=mx;j++){
                    ans[idx++]=mat[k-j][j];
                }
            }else{
                for(int j=mx;j>=mn;j--){
                    ans[idx++]=mat[k-j][j];
                }
            }
        }
        return ans;
    }
}

GO版本:

go 复制代码
func findDiagonalOrder(mat [][]int) []int {
    n,m:=len(mat),len(mat[0])
    ans:=make([]int,n*m)
    idx:=0
    // i+j=k  -> j=k-i;
    for k:=0;k<n+m-1;k++ {
        mn:=max(k-(n-1),0)
        mx:=min(k-0,m-1)
        if k%2==0 {
            for j:=mn;j<=mx;j++ {
                ans[idx]=mat[k-j][j]
                idx++
            }
        }else{
            for j:=mx;j>=mn;j-- {
                ans[idx]=mat[k-j][j]
                idx++
            }
        }
    }
    return ans
}
相关推荐
2401_8414956412 小时前
【操作系统】模拟真实操作系统核心功能的Java实现
java·操作系统·进程管理·系统调用·并发控制·中断处理·cpu调度
给大佬递杯卡布奇诺12 小时前
FFmpeg 基本数据结构 AVPacket分析
数据结构·c++·ffmpeg·音视频
南方的狮子先生12 小时前
【数据结构】从线性表到排序算法详解
开发语言·数据结构·c++·算法·排序算法·1024程序员节
派大星爱吃猫12 小时前
快速排序和交换排序详解(含三路划分)
算法·排序算法·快速排序·三路划分
焜昱错眩..12 小时前
代码随想录第四十八天|1143.最长公共子序列 1035.不相交的线 53. 最大子序和 392.判断子序列
算法·动态规划
程序员皮皮林12 小时前
Java 25 正式发布:更简洁、更高效、更现代!
java·开发语言·python
好家伙VCC12 小时前
**发散创新:AI绘画编程探索与实践**随着人工智能技术的飞速发展,AI绘
java·人工智能·python·ai作画
勇者无畏40412 小时前
基于 Spring AI Alibaba 搭建 Text-To-SQL 智能系统(前置介绍)
java·后端·spring·prompt·embedding
练习时长一年12 小时前
IDEA开发常用快捷键总结
java·ide·intellij-idea
程序猿编码12 小时前
Linux 文件变动监控工具:原理、设计与实用指南(C/C++代码实现)
linux·c语言·c++·深度学习·inotify