力扣第6题解：Z字型变换——Swift高手之道

2024-02-18 10:12:32

Z字型变换——Swift之道

力扣第6题：Z字型变换，看似简单，却蕴含着算法与编程的奥妙。本篇题解将带你领略Swift语言的魅力，以清晰易懂的讲解，带你步步攻克这道难题。

算法思想

Z字型变换的本质是将字符串按照一定规律排列，使其在视觉上呈现出Z字形的形态。为了实现这一效果，我们需要将字符串拆分成若干行，然后按照从上到下、从左到右的顺序进行排列。

Swift实现

func convert(_ s: String, _ numRows: Int) -> String {
    // 判断numRows的合法性
    guard numRows > 0 else { return "" }

    // 计算每个Z字形的宽度
    let width = 2 * numRows - 2

    // 初始化一个二维数组来存储Z字形
    var grid = Array(repeating: Array(repeating: " ", count: width), count: numRows)

    // 将字符串拆分成字符数组
    var chars = Array(s)

    // 定义行号和列号
    var row = 0
    var col = 0

    // 方向指示符，1表示向下，-1表示向上
    var direction = 1

    // 遍历字符串
    while chars.count > 0 {
        // 将字符添加到网格中
        grid[row][col] = chars.removeFirst()

        // 更新行号和列号
        if direction == 1 {
            row += 1
        } else {
            row -= 1
            col += 1
        }

        // 判断是否需要改变方向
        if row == numRows - 1 || row == 0 {
            direction *= -1
        }
    }

    // 将网格中的字符拼接成字符串
    var result = ""
    for row in grid {
        for char in row {
            if char != " " {
                result += char
            }
        }
    }

    return result
}