LeetCode(java)6. ZigZag Conversion

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

题目描述:给定一个字符串，然后将该字符串以行数为nRows的z型放置，然后按行读取并返回新的字符串。

例如：输入ABCDEFGHIJKLMNOPQRST，nRows为5时

排列如下：

应输出：AIQBHJPRCGKOSDFLNTEM

当nRows为2的时候比较形状比较特殊：

解题思路：

模拟已经按z型放置好了，则可计算出每行输出的字符在原字符串中的位置。

第0行中每个字符的位置为：0，(nRows - 1) * 2，(nRows - 1) * 4......

第1行-第n-2行中第 i 行的位置为： i ,  i + (nRows - i - 1) * 2 , i+(nRows - 1) * 2)......

最后一行：nRows - 1 , (nRows - 1)*2 , (nRows - 1)*3......

算法复杂度为O(n),leetcode通过时间为：8ms。

代码如下：

public class Solution {
    public String convert(String s, int nRows) {
        if (s == null || s.length() <= nRows || nRows <= 1) return s;
        StringBuffer sb = new StringBuffer();
        // 首行
        for (int i = 0; i < s.length(); i += (nRows - 1) * 2) {
            sb.append(s.charAt(i));
        }
        // 次行 —> 倒数第2行
        for (int i = 1; i < nRows - 1; i++) {
            for (int j = i; j < s.length(); j+= (nRows - 1) * 2) {
                sb.append(s.charAt(j));
                if (j + (nRows - i - 1) * 2 < s.length()) {
                    sb.append(s.charAt(j + (nRows - i - 1) * 2));
                }
            }
        }
        // 最后一行
        for (int i = nRows - 1; i < s.length(); i += (nRows - 1) * 2) {
            sb.append(s.charAt(i));
        }
        return sb.toString();
    }
}