博客专栏地址:https://blog.csdn.net/feng964497595/category_9848847.html
github地址:https://github.com/mufeng964497595/leetcode
题目描述
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
L C I R
E T O E S I I G
E D H N
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“LCIRETOESIIGEDHN”。
请你实现这个将字符串进行指定行数变换的函数。
思路解析
- 一开始想着找规律,后面发现其实好像没啥规律。。。
- 再仔细研究一下题目,发现直接照做就可以了,就是一道模拟题。搞个数组来存储每行的数据,每访问字符串的一个字符,就更新数组的下标即可。好像也没啥别的好说的了。。。感觉看代码更直白。。。
- 小插曲,代码中的offset最开始写的是offset = -offset,提交代码36ms,改为位运算offset = ~offset + 1,提交代码16ms,位运算大法好!
示例代码
class Solution {
public:
string convert(string s, int numRows) {
int len = s.length();
if (1 == numRows || len <= numRows) return s;
std::vector<std::string> vec;
vec.resize(numRows);
int offset = -1; // 偏移量,控制Z字走向
for (int i = 0, idx = 0; i < len; ++i) {
vec[idx] += s[i];
if (idx == numRows - 1 || idx == 0) {
offset = ~offset + 1; // 到边缘了,调个头
}
idx += offset;
}
std::string ans;
for (auto& str : vec) {
ans.append(str);
}
return ans;
}
};