6. Z 字形变换

问题

将一个给定字符串 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);

示例 1：
输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

分析

要么数学转换，要么模拟。这题可以通过设置循环量的正负来实现大模拟，十分巧妙。也就是按照Z字形存字符。
参考：https://leetcode.cn/problems/zigzag-conversion/solutions/21610/zzi-xing-bian-huan-by-jyd/

代码

class Solution {
public:string convert(string s, int numRows) {if (numRows < 2) {return s;}int s_n = s.size();vector<string> rows(numRows);int x = 0, step = -1;for (int i = 0; i < s_n; i++) {if (x == 0 || x == numRows-1) {step = -step;}rows[x].push_back(s[i]);x += step;}string res;for (const string &row : rows) {res += row;}return res;}
};