将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "LEETCODEISHIRING"
行数为 3 时,排列如下:
L C I R
E T O E S I I G
E D H N
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"LCIRETOESIIGEDHN"
。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L D R
E O E I I
E C I H N
T S G
我们可维护numRows个字符数组来存储每行的字符,按照规则,如果numRows为1直接返回s,否则就自上而下,从下往上周期性对每个字符数组添加字符。
class Solution {
public:
string convert(string s, int numRows) {
vector<vector<char>>p;
vector<char> a;
for(int i=0;i<numRows;i++)
p.push_back(a);
int count=0,i=0,flag=1;
if(numRows==1)return s;
while(count<s.size()){
if(1){
p[i].push_back(s[count++]);
if(flag)i++;
else i--;
if(i==numRows){
flag=0;
i-=2;
}
if(i==-1){
flag=1;
i+=2;
}
}
}
string sl;
for(int i=0;i<numRows;i++)
for(int j=0;j<p[i].size();j++)
sl.append(1,p[i][j]);
return sl;
}
};