[Leetcode #6]ZigZag Conversion 将字符串转换成ZigZag形式

原题地址：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

下面就是找个计算公式套进去就行了，第一层和最后一层的步长就是窗口长度window，中间第i层的第一步长为(window - 2 * i)，第二步长为(window - 第一步长)。

至此，分析完毕，所有需要的计算公式也已经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();
}
}