LeetCode——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

---

想法：完全的根据自己找的数学规律来，因为看到题目之后心中没有一种想用的数据结构，于是乎在纸上画了半天，感觉有一点规律，便动手把这所谓的规律实现了。

以示例2为例，将其中的"红色的V"看成一个整体"entity",易知一个entity由 2*numRows-2 个字母组成（其中numRows是题中提到的行数），最后一个"绿色的V"是因为字母数量有限，不能构成一个完成的entity。用count表示完整entity的个数，则count = s.length() / entity ;然后按照每行从左至右的顺序将其遍历一遍就OK了！

上代码：

class Solution {
    public String convert(String s, int numRows) {
        if(s == null || s.length() == 0)
            return "";
        //行数为1时直接返回s即可
        if(numRows == 1)
            return s;
        //计算entity和count
        int entity = 2 * numRows - 2;
        int count = s.length() / entity;
        //用StringBuffer来接收遍历的字母
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j <= count; j++) {
                //以下在 "end□" 之前是考虑竖线中的字母
                //这个条件判断是为了防止s.charAt(index)越界
                //即上面解释中提到的"绿色V"
                if(j * entity + i < s.length()){
                    sb = sb.append(s.charAt(j * entity + i));
                }
                //end□

                //以下在 "end△" 之前都是考虑斜线上的字母
                //因为第1行和第numRows行 斜线上没有字母
                if(i == 0 || i == numRows - 1){
                    continue;
                }
                //同样为了防止s.charAt(index)越界
                if((j + 1) * entity - i < s.length()){
                    sb.append(s.charAt((j + 1) * entity - i));
                }
                
                //end△
            }
        }
        return sb.toString();
    }
}

运行时间比较短，内存空间也比较少，自认为到目前为止做的较好的一道题吧，加油！

最后，各位路过的大佬，如果有更好的解法，欢迎指导，谢谢哦！