Z 字形变换

1、参考资料

https://leetcode-cn.com/problems/zigzag-conversion/

2、题目描述

题目要求

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

3、代码思路

由于最后输出的结果需要按行输出，所以我们使用 List<StringBuilder> rows 来存储结果，比如

L C I R E T O E S I I G E D H N

遍历完成后， rows 中存储的内容为

LCIR ETOESIIG EDHN

然后我们将 rows 中的内容，使用 StringBuilder 拼接，然后返回即可

---

关键点：

rowIndex 为集合 rows 的行索引，dir 用于记录方向，1 表示向下，-1 表示向上

初始化 rowIndex = 0，dir = 1，此时 rowIndex 从上往下走，在之后的步骤中，rowIndex 应该在增大和减小之间不断重复

当 rowIndex >= rows.size() - 1 时，说明 rowIndex 触底，应该反弹，之后 rowIndex 应该减小

当 rowIndex <= 0 时，rowIndex 触顶，应该反弹，之后 rowIndex 应该增大

4、代码实现

代码

/** * @ClassName ZConvertDemo * @Description TODO * @Author Heygo * @Date 2020/9/3 16:20 * @Version 1.0 */ public class ZConvertDemo { public static void main(String[] args) { Solution solution = new Solution(); String convertStr = solution.convert("LEETCODEISHIRING", 3); System.out.println(convertStr); } } class Solution { public String convert(String str, int numRows) { // Guard safe if (numRows == 1) { return str; } // 每一行都是一个 StringBuilder 对象 List<StringBuilder> rows = new ArrayList<>(); // 创建 ArrayList 中的 StringBuilder 对象 for (int i = 0; i < Math.min(str.length(), numRows); i++) { rows.add(new StringBuilder()); } int dir = 1; // 1 表示向下，-1 表示向上 int rowIndex = 0; // 行索引 // 遍历字符串中每个字符 for (char c : str.toCharArray()) { rows.get(rowIndex).append(c); // 将字符添加至对应行中 if (rowIndex >= rows.size() - 1) { // 行索引触底反弹 dir = -1; } else if (rowIndex <= 0) { // 行索引触底反弹 dir = 1; } rowIndex += dir; } // 拼接结果 StringBuilder result = new StringBuilder(); for (StringBuilder row : rows) { result.append(row); } return result.toString(); } }