[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();
    }
}