字符串题目：Z 字形变换

题目

标题和出处

标题：Z 字形变换

出处：6. Z 字形变换

难度

4 级

题目描述

要求

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

比如输入字符串为 $\text{"PAYPALISHIRING"}$，行数为 $\text{3}$ 时，排列如下：

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

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

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

$\text{string convert(string s, int numRows);}$

示例

示例 1：

输入：$\text{s = "PAYPALISHIRING", numRows = 3}$
输出：$\text{"PAHNAPLSIIGYIR"}$

示例 2：

输入：$\text{s = "PAYPALISHIRING", numRows = 4}$
输出：$\text{"PINALSIGYAHRPI"}$
解释：

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

示例 3：

输入：$\text{s = "A", numRows = 1}$
输出：$\text{"A"}$

数据范围

• $\text{1}\le \text{s.length}\le \text{1000}$
• $\text{s}$ 由英文字母（小写和大写）、$\text{‘,’}$ 和 $\text{‘.’}$ 组成
• $\text{1}\le \text{numRows}\le \text{1000}$

解法

思路和算法

首先考虑最简单的情形：当 $\text{numRows}=1$ 时，将字符串 $s$ 按照 Z 字形排列只有 $1$ 行，且字符顺序不变，因此在 Z 字形变换之后的字符串仍然是 $s$，直接返回 $s$。

当 $\text{numRows}\ge 2$ 时，Z 字形排列会有多行。在生成 Z 字形排列时，需要从左到右遍历字符串 $s$，对于每个字符，计算得到该字符位于 Z 字形排列的哪一行，然后将该字符拼接到所在行。

每个字符在 Z 字形排列中的行下标为从 $0$ 增加到 $\text{numRows}-1$，然后减少到 $0$，重复在该范围内变化。因此需要维护行下标 $\text{rowIndex}$ 和变化方向 $\text{direction}$，初始时 $\text{rowIndex}=0$，$\text{direction}=1$。遍历 $s$ 的每个字符时，将该字符拼接到下标为 $\text{rowIndex}$ 的行，然后将 $\text{rowIndex}$ 的值增加 $\text{direction}$，则更新后的 $\text{rowIndex}$ 为下一个字符应该拼接到的行的下标。由于行下标的取值范围是 $[0,\text{numRows}-1]$，因此当更新后的 $\text{rowIndex}$ 为 $0$ 或 $\text{numRows}-1$ 时，需要将 $\text{direction}$ 取相反数，表示下次行下标的更新需要往相反方向移动。

对 $s$ 遍历结束之后，即可得到 Z 字形排列的每一行，将每一行拼接之后，即可得到 Z 字形变换后的结果。

具体实现方面，由于涉及到字符串的拼接，因此使用 $\text{StringBuffer}$ 实现。创建长度为 $\text{numRows}$ 的 $\text{StringBuffer}$ 类型数组，该数组的每个元素对应 Z 字形排列的每一行。在拼接每一行得到最终答案时，同样使用 $\text{StringBuffer}$ 类型进行拼接。

还有一个可以优化的点。由于 Z 字形排列的每一行都至少有一个字符，因此 Z 字形排列的行数不会超过 $s$ 的长度。当 $s$ 的长度为 $n$ 时，Z 字形排列的行数为 $\min (\text{numRows},n)$，因此在生成 Z 字形排列之前，令 $\text{numRows}=\min (\text{numRows},n)$，可以达到优化的效果。

代码

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) {
            return s;
        }
        int length = s.length();
        numRows = Math.min(numRows, length);
        StringBuffer[] rows = new StringBuffer[numRows];
        for (int i = 0; i < numRows; i++) {
            rows[i] = new StringBuffer();
        }
        int rowIndex = 0, direction = 1;
        for (int i = 0; i < length; i++) {
            rows[rowIndex].append(s.charAt(i));
            rowIndex += direction;
            if (rowIndex == 0 || rowIndex == numRows - 1) {
                direction = -direction;
            }
        }
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < numRows; i++) {
            sb.append(rows[i]);
        }
        return sb.toString();
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是字符串 $s$ 的长度。创建 $\text{StringBuffer}$ 类型的数组并初始化元素需要 $O(\text{numRows})$ 的时间，生成 Z 字形排列和拼接得到 Z 字形变换后的结果各需要 $O(n)$ 的时间，由于会对 $\text{numRows}$ 预处理确保 $\text{numRows}\le n$，因此总时间复杂度是 $O(n)$。

• 空间复杂度：$O(n)$，其中 $n$ 是字符串 $s$ 的长度。Z 字形排列的结果需要 $O(n)$ 的空间存储，Z 字形变换后的结果也需要 $O(n)$ 的空间存储。由于 Java 中的 $\text{String}$ 类型的对象不可变，因此存储 Z 字形变换后的结果的 $O(n)$ 空间无法省略。