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

/**
 * Created by lxw, liwei4939@126.com on 2018/3/3.
 */
public class ZigZagConversion {

    public String convert(String s, int nRows){

        char[] ch = s.toCharArray();
        int len = ch.length;
        StringBuilder[] sb = new StringBuilder[nRows];

        for (int i = 0; i < sb.length; i++){
            sb[i] = new StringBuilder();
        }

        int i = 0;
        while (i < len){
            for (int index = 0; index < nRows && i < len; index++){
                sb[index].append(ch[i++]);
            }

            for (int idx = nRows - 2; idx >= 1 && i <len; idx--){
                sb[idx].append(ch[i++]);
            }
        }

        for (int idx = 1; idx < sb.length; idx++){
            sb[0].append(sb[idx]);
        }

        return sb[0].toString();
    }

    public static void main(String[] args){

        ZigZagConversion tmp = new ZigZagConversion();
        String str = "paypalishiring";
        String res = tmp.convert(str, 3);
        System.out.println(res);
    }
}