leetcode - 6.ZigZag Conversion

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

Solution:

  private StringBuilder sb = new StringBuilder();

    public String convert(String s, int numRows) {
        if (numRows == 1) {
            return s;
        }

        char[] chars = s.toCharArray();

        for (int i = 0; i < numRows; i++) {
            fullConvert(chars, numRows, i);
        }

        return sb.toString();
    }

    private void fullConvert(char[] chars, int numRows, int row) {
        int step = 2 * numRows - 2;
        for (int i = row; i < chars.length; i += step) {
            if (i % step == 0 || (i - numRows + 1) % step == 0) {
                sb.append(chars[i]);
            } else {
                int j = i + 2 * (numRows - row) - 2;
                sb.append(chars[i]);
                if (j < chars.length) {
                    sb.append(chars[j]);
                }
            }
        }
    }