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
分析:题目容易理解,第一反应是二维数组,循环填入就好了,空白地方用#,看了解析,下面是两种方法。
官方代码:
class Solution {
public:
string convert(string s, int numRows) {
if(numRows==1)return s;
vector<string> rows(min(numRows,int(s.size())));
int curRow = 0;
bool goingDown = false;
for(char c:s){
rows[curRow] += c;
if(curRow == 0 || curRow == numRows - 1)goingDown = !goingDown;
curRow += goingDown?1:-1;
}
string res;
for(string row:rows)res+=row;
return res;
}
};
其实就是几行字符串,第一行存一个字符串,第二行存一个,最后按照顺序合起来,一开始二维数组给忘记了其实不需要存放#空白,题目没有看清,用二维数组也很快,因为这个题规律很容易找,下面就是通过规律来存放字符串,定义一个字符串就可以了。
规律
class Solution {
public:
string convert(string s, int numRows) {
if (numRows == 1) return s;
int step = numRows * 2 - 2; // 间距
int index = 0;// 记录s的下标
int len = s.length();
int add = 0; // 真实的间距
string ret;
for (int i = 0; i < numRows; i++) // i表示行号
{
index = i;
add = i * 2;
while (index < len)//超出字符串长度计算下一层
{
ret += s[index]; // 当前行的第一个字母
add = step - add;// 第一次间距是step -2*i,第二次是2*i,
index += (i == 0 || i == numRows-1) ? step : add; // 0行和最后一行使用step间距,其余使用add间距
}
}
return ret;
}
};
规律就是0行和numRow-1行,两个字符间隔为numRows2-2,中间行为numRows2-2-2i和2i交替出现,i为行号。代码中add=step-add这行来实现的。
2.总结
这道题算数学题了吧,不过代码实现还是比较难。