ZigZag Conversion (LeetCode Algorithm Problem)

本题的要求是将一个普通字符串转化成二维图像下的zigzag形式，最后每行输出字符。
原题链接

---

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)

And then read line by line: “PAHNAPLSIIGYIR”

---

有一种直观的做法就是生成一个2维的vector，然后按照zigzag的规矩往这个2维vector里面填字母，最后顺序输出这个vector即可。
不过我更喜欢一个不生成额外空间的算法，就是找映射关系。直接将原string的每一个字母映射到目标string上就好了。
首先需要说明的是什么是一个循环。我们可以看到zigzag的一个循环就是一次直线往下+一次45°往上。再接下来便是第二次循环，如图。

那么在每一个循环里面，第一行和最后一行只输出1个字符，其他行都输出两个字符。输出的第一个字符当然是行数r，而第2个字符就是r+2*(n-r-1)，其中n指总行数。

附上cpp代码：

void sub_process(string& s1, string& s2, int cur_row, int num_rows) {
    int pos = cur_row;
    while (pos < s1.size()) {
        s2.push_back(s1[pos]);
        int next = pos + 2 * (num_rows - cur_row - 1);
        if (cur_row != 0 && cur_row != num_rows - 1 && next < s1.size()) {
            s2.push_back(s1[next]);
        }
        if (num_rows != 1) {
            pos += 2 * (num_rows - 1);
        }
        else {
            pos += 1;
        }
    }
}

class Solution {
public:
    string convert(string s, int numRows) {
        string result = string();
        for (int i = 0; i < numRows; i++) {
            sub_process(s, result, i, numRows);
        }
        return result;
    }
};