leetcode - 6. Z 字形变换

题目网址：https://leetcode-cn.com/problems/zigzag-conversion/

这个题目，第一次看到的时候，我想到的是，申请一个二维数组，然后遍历字符串，给对应的二维数组的字符赋值。

当时感觉有点复杂，然后就看了一下官方题解，静下心看了一阵，理解了思路，记录下来，后面面试准备是快速理解思路：

 

• 首先，题目只是要打印出z字形变换的字符串，并没有要求把空格之类的也打印出来，因此直接申请一个vector<string> rows，用来按行存放变换后的字符串；

即：原串："LEETCODEISHIRING"

最终结果：

那么：row[0] = LCIR; row[1] = ETOESIIG; row[2] = EDHN;

 

• 然后，根据z字形变换的特性，到rownum = 0的时候，向下移动，到rownum = max（输入参数2）的时候，向上移动。同时利用string的+运算，构造出最终的vector<string>。
• 最后，按行输出（遍历vector），得到最终结果。

 

我的题解：

class Solution {

public:

    string convert(string s, int numRows) {

 

        if (numRows == 1) return s;

 

        vector<string> rows(min(numRows, int(s.size()))); 

        int rownum = 0;

        bool movedown = false;

 

        for (char c : s) {

            rows[rownum] += c;

            if (rownum == 0 || rownum == numRows-1)

                movedown = !movedown;

            rownum += (movedown ? 1 : -1);

        }

 

        string ret;

        for (string str : rows)

        {

            ret += str;

        }

        return ret;

 

    }

};