矩阵逆时针打印golang

发布时间:2024-11-05 18:29:28

矩阵逆时针打印算法的实现与优化

在golang开发中,矩阵操作是一个常见的需求。矩阵逆时针打印是其中的一个经典问题,本文将讨论该问题的算法实现以及优化。

问题描述

给定一个二维矩阵,要求按照逆时针的顺序打印出所有元素。例如,对于如下的3x3矩阵:

1 2 3
4 5 6
7 8 9

应该打印出1、4、7、8、9、6、3、2、5的顺序。

算法实现

一个最直接的思路是按照层级逐层打印。定义四个边界,分别表示当前层的上边界top、下边界bottom、左边界left和右边界right。每次打印完一条边后,相应的边界缩小一格,并更新打印的方向。具体实现如下:

func printMatrix(matrix [][]int) []int {
    if len(matrix) == 0 {
        return nil
    }
    
    m, n := len(matrix), len(matrix[0])
    top, bottom, left, right := 0, m-1, 0, n-1
    var res []int
    
    for top <= bottom && left <= right {
        // 从左到右打印上边
        for i := left; i <= right; i++ {
            res = append(res, matrix[top][i])
        }
        top++
        
        // 从上到下打印右边
        for i := top; i <= bottom; i++ {
            res = append(res, matrix[i][right])
        }
        right--
        
        // 从右到左打印下边
        if top <= bottom {
            for i := right; i >= left; i-- {
                res = append(res, matrix[bottom][i])
            }
            bottom--
        }
        
        // 从下到上打印左边
        if left <= right {
            for i := bottom; i >= top; i-- {
                res = append(res, matrix[i][left])
            }
            left++
        }
    }
    
    return res
}

以上实现的时间复杂度为O(m * n),其中m为矩阵的行数,n为矩阵的列数。

算法优化

上述实现的空间复杂度为O(1),即不需要额外的存储空间。但是该实现对于一些特殊情况会存在冗余的计算。例如,当矩阵只有一行或一列时,前述算法仍然会进行无效的计算。

为了优化,我们可以引入一个visited数组,用于记录已经访问过的元素。遍历矩阵时,首先检查当前格子是否已经访问过,如果没有,则将其加入结果数组,并将visited对应位置置为true,否则直接进入下一轮遍历。具体实现如下:

func printMatrix(matrix [][]int) []int {
    if len(matrix) == 0 {
        return nil
    }
    
    m, n := len(matrix), len(matrix[0])
    top, bottom, left, right := 0, m-1, 0, n-1
    var res []int
    visited := make([][]bool, m)
    for i := range visited {
        visited[i] = make([]bool, n)
    }
    
    for top <= bottom && left <= right {
        // 从左到右打印上边
        for i := left; i <= right; i++ {
            if !visited[top][i] {
                res = append(res, matrix[top][i])
                visited[top][i] = true
            }
        }
        top++
        
        // 从上到下打印右边
        for i := top; i <= bottom; i++ {
            if !visited[i][right] {
                res = append(res, matrix[i][right])
                visited[i][right] = true
            }
        }
        right--
        
        // 从右到左打印下边
        if top <= bottom {
            for i := right; i >= left; i-- {
                if !visited[bottom][i] {
                    res = append(res, matrix[bottom][i])
                    visited[bottom][i] = true
                }
            }
            bottom--
        }
        
        // 从下到上打印左边
        if left <= right {
            for i := bottom; i >= top; i-- {
                if !visited[i][left] {
                    res = append(res, matrix[i][left])
                    visited[i][left] = true
                }
            }
            left++
        }
    }
    
    return res
}

优化后的空间复杂度为O(m * n),但由于引入了visited数组,所以此算法的时间复杂度仍然是O(m * n)。

总结

本文讨论了矩阵逆时针打印问题的算法实现与优化。根据不同的需求和数据规模,选择适合的算法可以提高程序的性能。在实际应用开发中,我们需要综合考虑时间复杂度、空间复杂度和代码可读性等因素,选择最合适的方案。

相关推荐