Leetcode: ZigZag Conversion

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R
And then read line by line: "PAHNAPLSIIGYIR"
Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);
convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".

一道数学题，参考了网上的思路，先计算一下每一个zig包含的字符个数，实际上是zigSize = 2*nRows – 2
然后一行一行的加s中的特定元素就行。
第一行和最后一行都只需要加一个字符，每一个zig，而且在s中的index间隔是zigSize。
中间每一行需要添加两个字符到结果中去。第一个字符同上；第二个字符的index就是zigSize – i。i是行index。

 需要注意的是，nRows == 1一定要单独讨论，否则rsize = 2*nRows - 2 == 0，循环没有增量会成为死循环。还有18-23行添加的顺序谁先谁后是有讲究的。

第三遍代码：361ms

public class Solution {
    public String convert(String s, int nRows) {
        if (s == null || s.length() == 0) return "";
        if (s.length() <= nRows || nRows == 1) return s;
        int rsize = 2 * nRows - 2;
        StringBuffer res = new StringBuffer();
        for (int i=0; i<nRows; i++) {
            if (i == 0 || i == nRows-1) {
                int t = 0;
                while (i + t * rsize < s.length()) {
                    res.append(s.charAt(i+t*rsize));
                    t++;
                }
            }
            else {
                int m = 0;
                while (i + m * rsize < s.length() || (rsize - i + m * rsize) < s.length()) {
                    if (i + m * rsize < s.length()) {
                        res.append(s.charAt(i+m*rsize));
                    }
                    if ((rsize - i + m * rsize) < s.length()) {
                        res.append(s.charAt(rsize-i+m*rsize));
                    }
                    m++;
                }
            }
        }
        return res.toString();
    }
}

 如果不分开i =0, i=nRows-1和其他case, 444ms, 比上面反而慢一点

public class Solution {
    public String convert(String s, int nRows) {
        if (s == null || s.length() == 0) return "";
        if (s.length() <= nRows || nRows == 1) return s;
        int rsize = 2 * nRows - 2;
        StringBuffer res = new StringBuffer();
        for (int i=0; i<nRows; i++) {
                int m = 0;
                while (i + m * rsize < s.length() || (rsize - i + m * rsize) < s.length()) {
                    if (i + m * rsize < s.length()) {
                        res.append(s.charAt(i+m*rsize));
                    }
                    if (i!=0 && i!=nRows-1 && ((rsize - i + m * rsize) < s.length())) {
                        res.append(s.charAt(rsize-i+m*rsize));
                    }
                    m++;
                }

        }
        return res.toString();
    }
}