力扣（LeetCode）试题6-Z字形变换 C++代码

这周课题组比较忙，牺牲了练题时间，该打该打。。。

中等题型，对于我还是挺难的 ಥ﹏ಥ ，不得不从答案找思路，这道题对于我的难点在于一次性生成numRows个容器/字符串

对于vector<>数据类型不是很熟悉，只好进行了遍历、拼接等操作，遍历了好几次，时间复杂度估计有点高，之后熟悉vector操作后再来改进。第一版代码如下：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution 
{
public:
    string convert(string s, int numRows) 
    {
        if (numRows == 0) return "数据不合法";
        {
            if (numRows == 1 || s == "") return s;
            else
            {
                int rows = min(numRows, int(s.size()));
                //一次生成rows个string类型的容器
                vector<string> str(rows);
                string result = "";
                int flag = -1;//控制步长的方向，+1还是-1
                int count = 0;//分配器，若为0或者rows-1,则反向

                for (int i = 0; i < s.length(); i++)
                {
                    if (count == 0 || count == rows - 1)
                    {
                        flag = -flag;
                    }
                    str[count].push_back(s[i]);//将第i个值存入str[count]容器中
                    count += flag;
                }

                int j = 1;//指向str[j]容器,依次取出字符，并拼接
                while (j < rows)
                {
                    for (int k = 0; k < str[j].size(); k++)
                    {
                        str[0].push_back(str[j][k]);
                    }
                    j += 1;
                }


                for (int q = 0; q < str[0].size(); q++)
                {
                    result += str[0][q];
                }
                return result;
            }
        }
    }
};

int main()
{
    Solution sol;
    string s = "";
    int n = 0;
    string result;
    result = sol.convert(s, n);
    for (int i = 0; i < result.size(); i++)
        cout << result[i];

    cout << endl;

    int u;
    cin >> u;

    return 0;
}