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
中每一个字符的下标进行如下变换:
按照题意,这个变换可以按照从上到下,再从下到上的顺序交替进行,伪代码如下:
- 初始化
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