[LeetCode - 字符串] 6. ZigZag Conversion

1 问题

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

2 分析

该问题可以这样理解：根据给定字符串S, 逐一将S中的字符变换到一个二维矩阵中特定的位置上（这个位置用(row,col)表示），再逐行将二维矩阵输出 。即, 对S中每一个字符的下标进行如下变换：

$$i\quad\to \quad(row,col)$$
按照题意，这个变换可以按照从上到下，再从下到上的顺序交替进行，伪代码如下：

• 初始化nRow个字符串，表示二维矩阵：stringMatrix = String[nRow]
• 令index表示给定字符中的数组下标，初始化为0。
• 令row表示二维矩阵中每一行的下标
• 进行如下循环，循环终止的条件是 index >= n:
  
  • row = 0
  • 进行内循环，表示从上到下对二维矩阵的一列进行复制, 终止条件是 row >= nRow
    
    • stringMatrix[row] += S[index]
    • row = row + 1
    • index = index + 1
  • row = nRow - 2
  • 进行内循环，表示从下到上对二维矩阵的一列进行复制, 终止条件是 row <= 1
    
    • stringMatrix[row] += S[index]
    • row = row - 1
    • index = index + 1
• 返回由stringMatrix组成的字符串

该算法遍历字符串一遍，时间复杂度是$O(n)$。

3 代码$^{[1]}$

public String convert(String s, int nRows) {
    char[] c = s.toCharArray();
    int len = c.length;
    StringBuffer[] sb = new StringBuffer[nRows];
    for (int i = 0; i < sb.length; i++) sb[i] = new StringBuffer();

    int i = 0;
    while (i < len) {
        for (int idx = 0; idx < nRows && i < len; idx++) // vertically down
            sb[idx].append(c[i++]);
        for (int idx = nRows-2; idx >= 1 && i < len; idx--) // obliquely up
            sb[idx].append(c[i++]);
    }
    for (int idx = 1; idx < sb.length; idx++)
        sb[0].append(sb[idx]);
    return sb[0].toString();
}

其中使用StringBuilder是为了增加字符串连接的效率。
[1]代码来源：discuss.leetcode.com