將一個(gè)給定字符串根據(jù)給定的行數(shù)，以從上往下、從左到右進(jìn)行 Z 字形排列。

比如輸入字符串為 "LEETCODEISHIRING" 行數(shù)為 3 時(shí)，排列如下：

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

之后，你的輸出需要從左往右逐行讀取，產(chǎn)生出一個(gè)新的字符串，比如："LCIRETOESIIGEDHN"。

請(qǐng)你實(shí)現(xiàn)這個(gè)將字符串進(jìn)行指定行數(shù)變換的函數(shù)：

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

解題思路

遍歷所有字符，維護(hù)兩個(gè)變量，當(dāng)前行號(hào)以及方向。當(dāng)當(dāng)前行號(hào)為最大值或者最小值時(shí)，變換方向。

實(shí)現(xiàn)代碼

//執(zhí)行用時(shí) :54 ms, 在所有 Java 提交中擊敗了27.85%的用戶
//內(nèi)存消耗 :67.7 MB, 在所有 Java 提交中擊敗了15.13%的用戶
class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) {
            return s;
        }
        
        int rows = Math.min(numRows, s.length());
        String[] lines = new String[rows];
        int curRow = 0;
        boolean down = false;
        for (int i = 0; i < s.length(); i++) {
            if (lines[curRow] == null) {
                lines[curRow] = s.substring(i, i + 1);
            } else {
                lines[curRow] += s.substring(i, i + 1);
            }
            
            if (curRow == 0 || curRow == rows - 1) {
                down = !down;
            }
            
            curRow = down ? curRow + 1 : curRow - 1;
        }
        
        String result = "";
        for (String line : lines) {
            result += line;
        }
        return result;
    }
}