leetcode 6.Z字形变换
题目描述
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
解题思路
该题考察的是对字符串的操作,将一个给定字符串根据给定的行数,以从==上往下、从左到右==进行 Z 字形排列。如果觉得题目描述的不清楚,看一下下面的动图1就可以很清楚的明白是怎么进行Z字形变形的啦。知道了转换的过程,按照上面的思路转换成代码就可以了。在这里我们使用的是建立一个二维的vector,更加变形规则不断的向每一行添加字符,最后再把每一个vector中的字符转成一整个字符串。代码如下:
class Solution {
public:
string convert(string s, int numRows) {
// 如果s是空,或者说numRows的数字小于等于1,就表示不需要转换,直接返回原始字符串
if(numRows <=1 || s.size() == 0){
return s;
}
vector<vector<char>> arr(numRows, vector<char>(0,0)); // 建立一个二维数组,初始化为空
string res; // 存储转换后的字符串
int len = s.size(); // 原始字符串的长度
bool flag = false; // 定义一个标志位,用于z行排列的变化
int row = 0; // 定义一个记录行号的变量,在该行下添加字符
for(int i=0; i<len; i++){
if(!flag){ // 如果flag为false,表示要从上到下添加字符
if(row == numRows-1){ // 如果row走到了最后一行,flag状态翻转,下次进行从左到右添加
flag = true;
arr[row].push_back(s[i]);
row--;
}
else{
arr[row].push_back(s[i]);
row++;
}
}
else{ // 如果flag为true,表示要从左到右添加字符
if(row == 0){ // 如果row走到了第0行,flag翻转,下次进行从上到下添加
flag = false;
arr[row].push_back(s[i]);
row++;
}
else{
arr[row].push_back(s[i]);
row--;
}
}
}
// 这个数组完成添加以后,逐行的合并字符串。
for(int i=0; i<numRows; i++){
for(int j=0; j<arr[i].size(); j++){
res += arr[i][j];
}
}
return res;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分享技术问题,我们一起学习进步
https://leetcode-cn.com/problems/zigzag-conversion/solution/zzi-xing-bian-huan-by-jyd/ ↩︎