【Leetcode】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

思路：

1）找规律。从行出发，我们可以很简单的发现半个$Z$字元素个数为$2\ast numRows-2$，以此为基准，得到下列规律

• 第$0$行，元素下标为$i(0)$，$i+2\ast numRows-2$， $i+2\ast （2\ast numRows-2）$，$i+3\ast （2\ast numRows-2）$
• 第$numRows-1$行，元素下标为$i(numRows-1)$， $i+2\ast numRows-2$, $i+2\ast (2\ast numRows-2)$, $i+3\ast (2\ast numRows-2)$
• 其余行元素下标，$i$，$2\ast numRows-2-i$, $2\ast numRows-2+i$, $2\ast (2\ast numRows-2)-i$
• 可以看出，当为第$0$行或$numRows-1$行时，该行的元素下标变化幅度都为$2\ast numRows-2$
• 当不为这两行时，同时计算两个元素，差为$2\ast numRows-2-2\ast i$

class Solution {
public:
    string convert(string s, int numRows) {
        if (s.size() <= 1 || numRows <= 1) 
            return s;
        string str;
        int n = s.size(), cyclelen = 2 * numRows - 2;
        for (int i = 0; i < numRows; ++i) {
            for (int j = 0; j + i < n; j += cyclelen) {
                str += s[j + i];
                //如果为其他行，则需计算半个Z字中的相邻元素，差值为cyclelen - 2 * i
                if (i != 0 && i != numRows - 1 && j + cyclelen - i < n) {
                    str += s[j + cyclelen - i];
                }
            }
        }
        return str;
    }
};

2）以行计算。将每一行的元素存入一个数组中，通过$bool$变量$next$判断存上一行还是下一行；当$next$为$true$时，则$row+1$，反之$row-1$，当为第$0$行或第$numRows-1$行时，需要反向，则令$next=!next$；遍历完之后，将元素按行存入$string$中返回

class Solution {
public:
    string convert(string s, int numRows) {
        if (numRows == 1)
            return s;
        vector<string> rows(max(numRows, (int)s.size()));
        int row = 0;
        bool next = false;
        for (char c : s) {
            rows[row] += c;
            if (row == 0 || row == numRows - 1)
                next = !next;
            row += next ? 1 : -1;
        }
        string str;
        for (string row : rows)
            str += row;
        return str;
    }
};