6. ZigZag Conversion

6. Z 字形变换

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

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

L   C   I   R
E T O E S I I G
E   D   H   N

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

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

string convert(string s, int numRows);

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

解法一：

//时间复杂度O(n), 空间复杂度O(1)
class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows == 1) return s;
        
        int n = s.size(), m = 2 * numRows - 2;
        string res;
        
        for(int i = 0; i < numRows; i++) {
            for(int j = i; j < n; j += m) {
                res += s[j];
                
                int temp = (numRows - i - 1) * 2;
                if(temp > 0 && temp < m && j + temp < n) res += s[j + temp];
            }
        }
        return res;
    }
};

思路：

仔细分析Z字型的排列，可以发现它具有周期性，其周期为m = 2 * numRows - 2。

1. 首先将字符串s分组，每组大小都为m（除了最后一组）；
2. 按行遍历。指针i指向当前行的行号（从0开始，numRows - 1是最后一行），指针j初始值为i，每次前进的步长为 m，直到j >= n时完成该行的遍历。然后i++遍历下一行；
3. 对遍历过程中j所指的每一个字符，追加至res（res += s[j];）（这是Z坚直线上的元素）。考虑还有斜线上的元素，需要将j加上一定的值temp = (numRows - i - 1) * 2，如果 0 < temp < m 且j + temp < n时，就代表j指针后还有一个处于斜线上的元素，也将其追加至res。
4. 遍历所有行都完成后，返回res。

2019/09/14 18:30