力扣题目解析--Z字形变换

原创于 2024-10-31 21:14:17 发布 · 1.8k 阅读

25 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

题目

将一个给定字符串 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"

提示：

1 <= s.length <= 1000
s 由英文字母（小写和大写）、',' 和 '.' 组成
1 <= numRows <= 1000

代码展示

class Solution {
public:
    string convert(string s, int numRows) {
         if (numRows == 1 || numRows >= s.size()) return s;

        vector<string> rows(min(numRows, int(s.size())));
        int n = s.size();
        int cycleLen = 2 * numRows - 2; // 一个完整的Z字形周期长度

        for (int i = 0; i < n; i++) {
            int pos = i % cycleLen;
            int row = numRows - abs(numRows - 1 - pos);
            rows[row - 1].push_back(s[i]);
        }

        string ret;
        for (const auto& row : rows) {
            ret += row;
        }
        return ret;
    }
};

代码的逐行详细解释

在Z字形排列中，当字符串按Z字形排列时，每个字符都会遵循一定的规律。假设我们有numRows行，那么从第一行到最后一行再回到第二行的路径形成了一个周期。

举例说明

假设我们有numRows = 3，那么Z字形排列如下：

P   A   H   N
A P L S I I G
Y   I   R

可以看到，字符的排列顺序是：

第一行的字符每隔2 * numRows - 2个字符出现一次；
第二行的字符每隔numRows - 2个字符出现一次；
最后一行的字符同样每隔2 * numRows - 2个字符出现一次。

对于第一行和最后一行，它们的周期是一样的，即每隔2 * numRows - 2个字符出现一次。这是因为：

字符从第一行跳转到最后一个字符需要经过numRows - 1个空格；
再从最后一个字符跳转回第一行需要再经过numRows - 1个空格；
因此总长度是numRows - 1 + (numRows - 1) = 2 * numRows - 2。

对于中间行，周期则是不同的。中间行的字符会交替出现在两行之间，其间隔会短一些。

数学解释

对于任意的numRows，我们可以这样理解：

每次从第一行跳到最后一行，再到第一行，构成了一个完整的周期；
跳跃的总长度是numRows - 1（从第一行跳到最后一行）加上numRows - 1（从最后一行跳回第一行）；
所以，一个完整的周期长度为2 * (numRows - 1)，即2 * numRows - 2。

cycleLen = 2 * numRows - 2 确定了字符在Z字形排列中的周期性规律，使得我们可以简单地通过模运算和除法来确定每个字符应该放置的位置。

解释 `pos = i % cycleLen`

i 是当前字符在原始字符串中的索引，cycleLen 是一个完整的Z字形周期的长度。i % cycleLen 计算的是当前字符在当前周期内的相对位置。

为什么使用模运算？

模运算 (%) 可以确保索引不会超出周期长度的范围。例如，在一个周期长度为 cycleLen 的情况下，如果 i 超过了 cycleLen，模运算就会给出 i 在 cycleLen 范围内的对应位置。这样可以保证我们在计算行数时不会出错。

如何确定行号？

接下来，我们需要根据 pos 来确定当前字符应该放在哪一行。由于Z字形排列的特性，我们可以根据 pos 来计算出当前字符应该位于哪一行。

对于 numRows 行的情况，周期长度为 2 * numRows - 2，因此：

当 pos 在 [0, numRows - 1) 范围内时，字符位于从上往下数的第 pos 行；
当 pos 在 [numRows, 2 * numRows - 2) 范围内时，字符位于从下往上数的第 2 * numRows - 2 - pos 行。

为了简化计算，我们可以统一使用一个公式来确定行号：

int row = numRows - abs(numRows - 1 - pos);

这个公式可以理解为：

如果 pos 在 [0, numRows - 1) 范围内，那么 numRows - 1 - pos 是正数，abs 不影响结果，row 就是 pos；
如果 pos 在 [numRows, 2 * numRows - 2) 范围内，那么 numRows - 1 - pos 是负数，abs 会取绝对值，row 就是 2 * numRows - 2 - pos。

在这个例子中，s 是 "PAYPALISHIRING"，numRows 是 3。

计算周期长度：cycleLen = 2 * numRows - 2 = 4。
遍历字符串：
- 对于 i = 0，pos = 0 % 4 = 0，row = 3 - abs(3 - 1 - 0) = 3 - 2 = 1，字符 'P' 放在第一行。
- 对于 i = 1，pos = 1 % 4 = 1，row = 3 - abs(3 - 1 - 1) = 3 - 1 = 2，字符 'A' 放在第二行。
- 对于 i = 2，pos = 2 % 4 = 2，row = 3 - abs(3 - 1 - 2) = 3 - 0 = 3，字符 'Y' 放在第三行。
- 对于 i = 3，pos = 3 % 4 = 3，row = 3 - abs(3 - 1 - 3) = 3 - 1 = 2，字符 'P' 放在第二行。
- 以此类推，直到字符串遍历结束。

最终，rows 中存储了每一行的字符，将这些字符拼接起来就得到了Z字形排列后的字符串。

目标

我们的目标是确定当前字符应该放置在哪一行。我们需要一个公式来根据当前字符在周期内的位置（即pos）来计算出它所在的行数。

分析

对于一个Z字形排列，我们可以观察到以下规律：

当字符处于周期的前半段时（即pos小于numRows - 1），它们从上到下依次放置；
当字符处于周期的后半段时（即pos大于等于numRows - 1），它们从下往上依次放置。

公式推导

前半段（`pos` < `numRows - 1`）

对于前半段，pos表示的是从第一行开始的行数，因此：

当pos为0时，字符应该放置在第一行；
当pos为1时，字符应该放置在第二行；
……
当pos为numRows - 1时，字符应该放置在最后一行。

因此，对于前半段，行号row就是pos。

后半段（`pos` ≥ `numRows - 1`）

对于后半段，我们需要从最后一行开始往回计数，因此：

当pos为numRows时，字符应该放置在倒数第二行；
当pos为numRows + 1时，字符应该放置在倒数第三行；
……
当pos为2 * numRows - 2时，字符应该放置在第一行。

因此，对于后半段，行号row应该是2 * numRows - 2 - pos。

统一公式

为了简化代码，我们可以通过一个统一的公式来计算行号row，无论pos是在前半段还是后半段。

int row = numRows - abs(numRows - 1 - pos);

公式解析

numRows - 1 - pos：这个表达式给出了一个数值，这个数值的正负取决于pos是处于前半段还是后半段。
- 当pos在前半段时，numRows - 1 - pos为正数；
- 当pos在后半段时，numRows - 1 - pos为负数。
abs(numRows - 1 - pos)：取上述表达式的绝对值，无论pos处于哪个阶段，这个表达式总是非负的。
numRows - abs(numRows - 1 - pos)：这个表达式确保了row总是有效的行号。
- 当pos在前半段时，row就是pos；
- 当pos在后半段时，row就是2 * numRows - 2 - pos。

示例

假设numRows = 3，周期长度cycleLen = 2 * numRows - 2 = 4，那么对于pos的不同值，row的计算如下：

pos = 0：row = 3 - abs(3 - 1 - 0) = 3 - 2 = 1 （第一行）
pos = 1：row = 3 - abs(3 - 1 - 1) = 3 - 1 = 2 （第二行）
pos = 2：row = 3 - abs(3 - 1 - 2) = 3 - 0 = 3 （第三行）
pos = 3：row = 3 - abs(3 - 1 - 3) = 3 - 1 = 2 （第二行）

通过这个公式，我们可以统一处理前半段和后半段的情况，从而正确地计算出行号row。

string ret;
for (const auto& row : rows) {
    ret += row;
}
return ret;

代码作用

这段代码的目的是将之前按照Z字形排列分配到各个行的字符重新组合成一个新的字符串。具体来说：

初始化结果字符串：
```
string ret;
```
创建一个空字符串ret，用于存放最终的结果字符串。
遍历所有行：
```
for (const auto& row : rows) {
    ret += row;
}
```
这里使用了范围基础的for循环来遍历rows向量中的每个字符串。rows是一个包含每一行字符的vector<string>。
拼接字符串：
```
ret += row;
```
对于rows中的每一个字符串row，将其追加到结果字符串ret的末尾。这样，每一行的字符都会依次被加入到ret中。
返回结果字符串：
```
return ret;
```
返回最终的结果字符串ret，它包含了按照Z字形排列规则重新组织后的字符序列。

`const auto& row : rows` 解释

这段代码是一个范围基础的for循环（range-based for loop），它是C++11引入的一种更加简洁和易读的方式来遍历容器中的元素。

语法解释

const：
- const关键字表明在此循环中，row是一个常量引用，意味着你不能通过row来修改原始容器中的元素。这样做可以提高效率，因为不需要复制元素。
auto：
- auto关键字用来自动推断row的类型。编译器会根据rows容器中的元素类型来自动确定row的类型。这样可以避免手动指定类型，使代码更简洁。
&：
- &表示row是一个引用（reference）。这意味着row直接引用容器中的元素，而不是复制它们。这样做可以避免在每次循环时复制元素，提高性能。
row：
- row是循环体中使用的变量名，代表每次迭代时当前元素的引用。
rows：
- rows是我们要遍历的容器（在这里是一个std::vector<std::string>）。

逐步解释

初始化：
- 在每次循环开始之前，row会被初始化为rows容器中的下一个元素的引用。
循环条件：
- 循环会一直执行，直到所有的元素都被处理完毕。
循环体：
- 每次迭代时，row代表当前元素的引用。在这个例子中，row是rows中的一个字符串。
更新结果字符串：
- ret += row; 这一行代码将当前的字符串row追加到结果字符串ret的末尾。

示例

假设rows是一个包含三个字符串的向量：{"PAHN", "YP", "ALSIGYIR"}。

第一次迭代：
- row引用rows[0]，即"PAHN"。
- ret += row; => ret = "PAHN"。
第二次迭代：
- row引用rows[1]，即"YP"。
- ret += row; => ret = "PAHNY"。
第三次迭代：
- row引用rows[2]，即"ALSIGYIR"。
- ret += row; => ret = "PAHNYALSIGYIR"。