原题地址:https://leetcode.com/problems/zigzag-conversion/
题目要求是:把一个字符串转换成ZigZag形式,返回按行读的结果。例如"PAYPALISHIRING"写成ZigZag形式如下:(估计是Paypal的面试题?)
需要返回按行读取的结果“PAHNAPLSIIGYIR”,如下图:
我们把字母全部替换成它们对应的下标值,看看有什么规律:
是不是感觉是一个长度为4的窗口在移动?3层的有点特殊,我们再看看4层的情况是什么样的:
这次窗口长度变成6了,以此类推,N层ZigZag的窗口长度为2 * (N - 1)。再看看每一层的分布是怎么样的:
可以看出,除了第一层和最后一层只有一个固定步长外,其他层都有两个步长交替出现。每一层移动的步长情况如下:
- 第1层的步长为6
- 第2层第一步长为4,第二步长为(6 - 4 = 2)
- 第3层第一步长为2,第二步长为(6 - 2 = 4)
- 第4层的步长为6
至此,分析完毕,所有需要的计算公式也已经ready,可以写代码了:
public class Solution {
public String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
StringBuffer sb = new StringBuffer();
int window = 2 * (numRows - 1);
int idx = 0;
int offset = 0;
for (int i = 0; i < numRows; i++) {
idx = i;
offset = window - 2 * i;
while (idx < s.length()) {
sb.append(s.charAt(idx));
if (offset == 0) {
// for the first and last row
offset = window;
}
idx += offset;
offset = window - offset;
}
}
return sb.toString();
}
}