LeetCode 146 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".

分析：

同一行两个字母相隔的距离是可以算出来的。

每一个周期的长度是 size = 2*nRows - 2，

中间行需要加多一个字母，这个字母记录同行上一个字母的距离是 size - 2*i.

public class Solution {
    public String convert(String s, int nRows) {
        if(s==null || s.length()==0 || nRows<=0)
            return "";
        if(nRows==1)
            return s;
        StringBuilder res = new StringBuilder();
        //一个折回周期的长度
        int size=2*nRows-2;
        for(int i=0; i<nRows; i++){
            //每个折回周期必然有一个数
            for(int j=i; j<s.length(); j+=size){
                res.append(s.charAt(j));
                //如果不是第0行和最后一行，则中间还要加一个数，距离为size-2*i
                if(i!=0 && i!=nRows-1 && j+size-2*i<s.length())
                    res.append(s.charAt(j+size-2*i));
            }
        }
        return res.toString();
    }
}