每日一题6

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

方法一：二维矩阵，写的有点痛苦了

class Solution {

public:

    string convert(string s, int numRows) {

        int len = s.length();

        if (numRows == 1 || numRows >= len) {

            return s;

        }

        int t = numRows * 2 - 2;

        int c = (len + t - 1) / t * (numRows - 1);

        vector<string> mat(numRows, string(c, 0));

        for (int i = 0, x = 0, y = 0; i < len; ++i) {

            mat[x][y] = s[i];

            if (i % t < numRows - 1) {

                ++x; // 向下移动

            } else {

                --x;

                ++y; // 向右上移动

            }

        }

        string ans;

        for (auto &row : mat) {

            for (char ch : row) {

                if (ch) {

                    ans += ch;

                }

            }

        }

        return ans;

    }

};