leetcode第六题--ZigZag Conversion

Problem：

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

就是将一个字符串按照“之”字型表示后，按每行组合后输出新字符串。

在白头翁的博客里学习到了O(n)的算法。

就是创建nRows个string，将s对应坐标拷贝到每个string中，利用pushback操作。最后append连接输出。代码如下：

class Solution {
public:
    string convert(string s, int nRows)
    {
        if (nRows <= 1 || s.length() < 2)
            return s;  

        string *s_arr = new string[nRows];
        int nCount = 2 * (nRows - 1);  // 每个循环的轮回 也就是 2倍的nRows - 2

        for (int i = 0; i < s.length(); ++i)
        {
            s_arr[nRows - 1 -abs(nRows - 1 - (i%nCount))].push_back(s[i]);
        }  

        string str_result;
        for (int i = 0; i < nRows; ++i)
            str_result.append(s_arr[i]);  

        return str_result;
    }
};

坐标对应是关键，数学逻辑思维确实要强。可以举个四行的例子试试。