6. Z 字形变换

一、问题

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

二、解题思路

• 如果只有 1 行或者字符串的长度小于等于行数，那么直接返回原字符串。
• 对于超过 1 行的情况，我们可以把字符串按照“Z 字形”从上到下再从下到上依次排放。
• 具体来说：
  • 使用一个数组 rows 来存储每一行的字符串。
  • 遍历字符串，逐个字符放入对应的行。遍历时通过一个指针 curRow 记录当前的行数，并使用一个变量 goingDown 控制方向（向下或向上）。
  • 当到达最后一行时，反向往上走；到达第一行时，反向往下走。
• 遍历完字符串后，将 rows 数组中的所有字符串依次拼接，形成最终结果。

三、代码

public class Solution {
    /**
     * 将字符串 s 按照 Z 字形排列，并返回排列后的字符串。
     *
     * @param s       输入字符串
     * @param numRows 行数
     * @return 按照 Z 字形排列后的字符串
     */
    public String convert(String s, int numRows) {
        // 如果行数小于等于 1，直接返回原字符串
        if (numRows <= 1) {
            return s;
        }

        // 创建一个 StringBuilder 数组，用于存储每一行的字符
        StringBuilder[] rows = new StringBuilder[numRows];
        // 初始化每个 StringBuilder
        for (int i = 0; i < numRows; i++) {
            rows[i] = new StringBuilder();
        }

        int curRow = 0; // 当前行索引
        boolean goingDown = false; // 是否向下移动

        // 遍历字符串 s 的每个字符
        for (char c : s.toCharArray()) {
            rows[curRow].append(c); // 将字符添加到当前行

            // 如果当前行是第一行或最后一行，改变方向
            if (curRow == 0 || curRow == numRows - 1) {
                goingDown = !goingDown;
            }

            // 根据方向，更新当前行索引
            curRow += goingDown ? 1 : -1;
        }

        // 将所有行的内容拼接成一个字符串
        StringBuilder result = new StringBuilder();
        for (StringBuilder row : rows) {
            result.append(row);
        }

        return result.toString(); // 返回结果
    }
}

四、时间和空间复杂度

• 时间复杂度: O(n)，其中 n 是字符串 s 的长度。我们遍历了每个字符，将它们放入 rows 数组的不同行，并最终将行的字符串拼接。
• 空间复杂度: O(n)，我们使用了 StringBuilder 数组来存储每行的字符，数组的大小与输入字符串的长度成正比。