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"
示例 2:
输入:s = "PAYPALISHIRING", numRows = 4
输出:"PINALSIGYAHRPI"
解释:
P I N
A L S I G
Y A H R
P I
示例 3:
输入:s = "A", numRows = 1
输出:"A"
思路:
创建数组z,用来存储z型的每一行的内容,然后将z的每一行内容相加输出
用字符串数组比用vector更快。
1、C++__自写
也可将vector,改为string数组
class Solution {
public:
string convert(string s, int numRows) {
vector<string> z(numRows,"");
string res="";
int n = s.length();
int j=1;
int c = 2*numRows-2;
if(numRows==1)
return s;
for(int i=0;i<n;i++){
if(i%c<numRows && i%c!=0){
j++;
}
else{
j--;
}
z[j]+=(s[i]);
}
for(auto each :z ){
res+=each;
}
return res;
}
};
2、C++——柳
class Solution {
public:
string convert(string s, int numRows) {
string z[numRows];
string res="";
int n = s.length();
int i=0;
while(i<n){
for(int j=0 ;j<numRows && i<n;j++){
z[j]+=s[i];
i++;
}
for(int k=numRows-2; k>0 && i<n;k--){
z[k]+=s[i];
i++;
}
}
for(auto each :z )
res+=each;
return res;
}
};
3、python
class Solution:
def convert(self, s: str, numRows: int) -> str:
n = len(s)
z = ['']*numRows
res=''
i=0
while(i<n):
for j in range(numRows):
if(i<n):
z[j]+=s[i]
i+=1
for k in range(numRows-2,0,-1):
if(i<n):
z[k]+=s[i]
i+=1
for each in z:
res+=each
return res