[LeetCode] #6 Z 字形变换

[LeetCode] #6 Z 字形变换

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

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

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

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

string convert(string s, int numRows);

输入：s = "PAYPALISHIRING", numRows = 4

输出："PINALSIGYAHRPI"

解释：

使用List<StringBuilder>来模拟Z形变换后的每一行，依次递增存到每一行，第一个第一行、第二个第二行.......当row等于0或numRows-1时，转向变成依次递减，第四个第四行、第五个第三行.....

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) return s;
        List<StringBuilder> list = new ArrayList<>();
         for (int i = 0; i < Math.min(numRows, s.length()); i++)
            list.add(new StringBuilder());
        boolean flag = false;
        int row = 0;
        for(char ch: s.toCharArray()){
            list.get(row).append(ch);
            if(row == 0 || row == numRows-1) flag = !flag;
            if(flag) row++;
            else row--;
        }
        StringBuilder res = new StringBuilder();
        for (StringBuilder str: list) res.append(str);
        return res.toString();
    }
}

也可以找规律

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) return s;
        StringBuilder res = new StringBuilder();
        int cycleLen = 2 * numRows - 2;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < s.length(); j += cycleLen) {
                res.append(s.charAt(j + i));
                if (i != 0 && i != numRows - 1 && j + cycleLen - i < s.length())
                    res.append(s.charAt(j + cycleLen - i));
            }
        }
        return res.toString();
    }
}

知识点：无

总结：无