Leecode_6.Z 字形变换

Leecode_6.Z 字形变换

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);

示例

输入:s = "PAYPALISHIRING", numRows = 3
输出:"PAHNAPLSIIGYIR"

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