6. ZigZag Conversion (M)

最新推荐文章于 2022-10-11 21:03:23 发布

原创最新推荐文章于 2022-10-11 21:03:23 发布 · 102 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

202 篇文章

订阅专栏

本文介绍了一种字符串处理算法，即将输入字符串按指定行数进行ZigZag（之字形）排列，并按行读取输出。文章提供了具体实例说明及Java实现代码。

ZigZag Conversion (M)

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

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

And then read line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string s, int numRows);

Example 1:

Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"

Example 2:

Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"

Explanation:

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

题意

给定字符串，将其按照指定行数进行之字形的重新排列，按顺序输出新图形的每一行字符。

思路

直接模拟字符重排的过程，按顺序将每一个字符存入对应行号的字符串中，注意在第一行和最后一行处要改变行号变化的方向。

代码实现

class Solution {
    public String convert(String s, int numRows) {
        // 特殊情况直接返回字符串
        if (numRows == 1) {
            return s;
        }
        
        // 分别保存重排后的每一行
        List<StringBuilder> sbs = new ArrayList<>();
        for (int i = 0; i < numRows; i++) {
            sbs.add(new StringBuilder());
        }
        
        // row指定当前字符对应的行号
        int row = 0;
        // flag用于判断行号增减方向
        boolean flag = false;
        // 按顺序将字符保存到对应的行字符串中
        for (char c : s.toCharArray()) {
            sbs.get(row).append(c);
            // 在第一行和最后一行要变换行号增减方向
            if (row == 0 || row == numRows - 1) {
                flag = !flag;
            }
            if (flag) {
                row++;
            } else {
                row--;
            }
        }
        
        StringBuilder res = new StringBuilder();
        for (StringBuilder sb : sbs) {
            res.append(sb);
        }
        return res.toString();
    }
}