6. Z 字形变换

ゝ一世哀愁。 2022-12-29 09:39 169阅读 0赞

# [6. Z 字形变换][6. 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();
        }
    }

[6. Z]: https://leetcode-cn.com/problems/zigzag-conversion/