LeetCode Java First 400 题解-006

ZigZag Conversion    Medium

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

public String convert(String s, int numRows) {

    char[] c = s.toCharArray();

    int len = c.length;

    StringBuilder[] sb = new StringBuilder[numRows];

    for (int i = 0; i < sb.length; i++)

        sb[i] = new StringBuilder();

    int i = 0;

    while (i < len) {

        for (int idx = 0; idx < numRows && i < len; idx++)

            sb[idx].append(c[i++]);

        for (int idx = numRows - 2; idx >= 1 && i < len; idx--)

            sb[idx].append(c[i++]);

    }

    for (int idx = 1; idx < sb.length; idx++)

        sb[0].append(sb[idx]);

    return sb[0].toString();

}

思路：正反反复读即可把ZigZag读出，存入StringBuilder（每行一个），然后按顺序拼接StringBuilder。注意反向时需要忽略首尾行。