LeetCode笔记-0006.Z 字形变换

6. Z 字形变换
难度：中等
来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目

将一个给定字符串 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：

输入：s = “PAYPALISHIRING”, numRows = 3
输出：“PAHNAPLSIIGYIR”

示例 2：

输入：s = “PAYPALISHIRING”, numRows = 4
输出：“PINALSIGYAHRPI”

示例 3：

输入：s = “A”, numRows = 1
输出：“A”

提示：

• 1 <= s.length <= 1000
• s 由英文字母（小写和大写）、’,’ 和 ‘.’ 组成
• 1 <= numRows <= 1000

求解

解法一：按行访问

分析

Z型字符串作为基准单元				每一行相邻两个元素索引的差值
A			G	6
B		F	H	4  2
C	E		I	2  4
D			J	6

从一个Z进行分析（numRows=4为例），一个Z范围内可以得出以下结论：

• 每一行的元素索引差值之和（interval）均为6（即 interval = 2*n-2）；
• 第一行和最后一行只有两个元素，其他行有三个元素；
• 每行前两个元素的索引差值递减2（最后一行除外）；
• 第一行元素的索引为[0, interval, 2*interval …]；
• 最后一行元素的索引为[interval-1, 2*interval-1,…]；
• 中间行的元素索引为[nowrow-1, nowrow-1+interval-2*(nowrow-1), nowrow-1 + interval, …]；

Java实现

public class Solution {
    public String convert(String s, int numRows) {
        // s为空/s的长度小于等于numRows，结果仍是s
        if (s == null || s.length() == 0 || s.length() <= numRows || numRows == 1) {
            return s;
        }
        int length = s.length();
        char[] sChars = s.toCharArray();
        StringBuilder stringBuilder = new StringBuilder();
        int interval = 2 * numRows - 2;
        for (int i = 1; i <= numRows; i++) {
            // 每个Z，第一行和最后一行两个元素间隔为（2 * numRows - 2）
            if (i == 1) {
                // 第一行，为 sChars[0]、sChars[interval]、sChars[2*interval]、...
                for (int j = 0; j < length; j += interval) {
                    stringBuilder.append(sChars[j]);
                }
                continue;
            }
            if (i == numRows) {
                // 最后一行，为 sChars[numRows-1]、sChars[numRows-1 + interval]、...
                for (int j = numRows - 1; j < length; j += interval) {
                    stringBuilder.append(sChars[j]);
                }
                continue;
            }
            // 中间的每行三个元素间隔和为（2 * numRows - 2），且逐行递减
            int count = 1;
            for (int j = i - 1; j < length; ) {
                stringBuilder.append(sChars[j]);
                // 计算下一次间隔
                if (count % 2 == 0) {
                    // 偶数
                    j += 2 * (i - 1);
                } else {
                    j += interval - 2 * (i - 1);
                }
                count++;
            }
        }
        return stringBuilder.toString();
    }
}

复杂度分析

• 时间复杂度：O(n)，sChars每个元素被访问一次；
• 空间复杂度：O(n)；

解法二：按行访问

分析

Z型字符串作为基准单元				每一行相邻两个元素索引的差值
A			G	6
B		F	H	4  2
C	E		I	2  4
D			J	6

以A-F、G-L（V字型）为基准单元进行分析，一个V的范围内可以得出以下结论nowRow表示当前行，1 <= nowRow <= numRows：

• 第一行和最后一行只有一个元素，其他行有两个元素；
• 第一行元素的索引为k * interval(其中k=0,1,2,...)；
• 最后一行元素的索引为(interval - 1) + k * interval(其中k=0,1,2,...)；
• 中间行第一个元素索引为(nowRow - 1) + k * interval(其中k=0,1,2,...)，第二个元素索引为(nowRow - 1) + (k + 1) * interval - 2(nowRow - 1)，即：(k + 1) * interval - (nowRow - 1) (其中k=0,1,2,...)

Java实现

public class Solution {
    public String convert(String s, int numRows) {
        if (s.length() <= numRows || numRows == 1) {
            return s;
        }
        // 初始化数据
        int n = s.length();
        int interval = 2 * numRows - 2;
        char[] sChars = s.toCharArray();
        StringBuilder stringBuilder = new StringBuilder();
        // 逐行循环
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < n; j += interval) {
                stringBuilder.append(sChars[j + i]);
                if (i != 0 && i != numRows - 1 && j + interval - i < n) {
                    stringBuilder.append(sChars[j + interval - i]);
                }
            }
        }
        return stringBuilder.toString();
    }
}

复杂度分析

• 时间复杂度：O(n)，sChars每个元素被访问一次；
• 空间复杂度：O(n)；

解法三：按行排序

分析

A			G
B		F	H
C	E		I
D			J

初始化一个List（用于存储每行的字符串），根据Z字型，可以逐个遍历字符串，将字符串放到对应的行中（类似二维数组）。

官方题解中用到了Math.min(numRows, s.length())，以此来避免遍历空行，其实可以通过s.length() <= numRows直接返回s即可，更容易理解。

Java实现

public class Solution {
    public String convert(String s, int numRows) {
        if (s.length() <= numRows || numRows == 1) {
            return s;
        }
        int n = s.length();
        char[] sChars = s.toCharArray();
        int currentRow = 0;
        // 表示从上往下还是从下往上走Z字型，true表示从上往下
        boolean isDown = false;
        // 类似于二维数组，遍历每个字符，放到对应的行，最终再逐行拼接
        List<StringBuilder> resultList = new ArrayList<>(numRows);
        for (int i = 0; i < numRows; i++) {
            resultList.add(new StringBuilder());
        }
        for (int i = 0; i < n; i++) {
            // 放到对应行的StringBuilder中
            resultList.get(currentRow).append(sChars[i]);
            // 遍历到第一行或者最后一行，开始转向
            if (currentRow == 0 || currentRow == numRows - 1) {
                isDown = !isDown;
            }
            // 根据走向修改当前行数
            currentRow += isDown ? 1 : -1;
        }
        StringBuilder result = new StringBuilder();
        resultList.forEach(e -> result.append(e.toString()));
        return result.toString();
    }
}

复杂度分析

• 时间复杂度：O(n)，sChars每个元素被访问一次；
• 空间复杂度：O(n)；